PyRank
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About

Xla Python Packages

Python packages with the GitHub topic xla. Sorted by relevance, with stars and monthly downloads.
pytorch
torch-xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

101K 3K 574
avitai
datarax

A Differentiable Data Pipeline Framework for JAX

5K 1 0
DifferentiableUniverseInitiative
jaxdecomp

JAX reimplementation and bindings for the NVIDIA cuDecomp library

2K 56 2
mpi4jax
mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

2K 525 32
clashluke
revlib

Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload

888 132 6
Microrobotics-Simulation-Framework
maddening

Modular Automatic Differentiation and Data-Enhanced Neural-network INteracting Graph (MADDENING). A pure JAX end-to-end autodifferentiable graph based framework for multiphysics simulation.

728 1 0
paolodelia99
tf-q-finance

High-performance TensorFlow library for quantitative finance.

589 1 0
felafax
felafax

felafax

440 571 39
bahremsd
tmmax

TMMax: High-performance modeling of multilayer thin-film structures using transfer matrix method with JAX

286 35 7
bahremsd
katmer

katmer is a powerful library for optimizing the design of optical thin films using automatic differentiation via JAX and Equinox, enabling efficient and accurate inverse design solutions.

81 11 3
    • Data from PyPI, GitHub, ClickHouse, and BigQuery