PyPI · Go · crates.io
wandb/wandb
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
MIT27 серп. 2026 р. · метрики 2.10.0
Усі теги
Тег каталогу
#jax
Усі репозиторії публічного реєстру з цим тегом — із тем GitHub або ключових слів, які публікують їхні реєстри пакетів. Здоров'я вимірюється за тією ж версіонованою методологією, що й решта реєстру.
46 записів
PyPI
PennyLaneAI/pennylane
PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.
Apache-2.013 серп. 2026 р. · метрики 2.10.0
PyPI
huggingface/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
Apache-2.012 серп. 2026 р. · метрики 2.10.0
PyPI
keras-team/keras
Deep Learning for humans
Apache-2.04 серп. 2026 р. · метрики 2.10.0
PyPI
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Apache-2.029 серп. 2026 р. · метрики 2.10.0
PyPI
iree-org/iree
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Apache-2.028 серп. 2026 р. · метрики 2.10.0
PyPI
keras-team/keras-hub
Pretrained model hub for Keras 3.
Apache-2.04 вер. 2026 р. · метрики 2.10.0
PyPI
HIPS/autograd
Efficiently computes derivatives of NumPy code.
MIT28 серп. 2026 р. · метрики 2.10.0
PyPI · crates.io
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
Apache-2.027 серп. 2026 р. · метрики 2.10.0
PyPI
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Apache-2.05 серп. 2026 р. · метрики 2.10.0
PyPI
google/grain
Library for reading and processing ML training data.
Apache-2.021 лип. 2026 р. · метрики 2.10.0
PyPI
pyro-ppl/numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Apache-2.013 серп. 2026 р. · метрики 2.10.0
PyPI
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Apache-2.030 лип. 2026 р. · метрики 2.10.0
PyPI
google/etils
Collection of eclectic utils for python.
Apache-2.027 серп. 2026 р. · метрики 2.10.0
PyPI
arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
MIT28 серп. 2026 р. · метрики 2.10.0
PyPI
astro-informatics/s2fft
S2FFT: Differentiable and accelerated spherical transforms
MIT16 серп. 2026 р. · метрики 2.10.0
PyPI · npm
magenta/magenta-realtime
Magenta RealTime 2: An Open-Weights Live Music Model
Apache-2.018 лип. 2026 р. · метрики 2.10.0
PyPI
tensorflow/datasets
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Apache-2.028 серп. 2026 р. · метрики 2.10.0
PyPI
google/torchax
torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytorch syntax together when authoring ML programs, and run it in every hardware JAX can run.
Apache-2.019 лип. 2026 р. · метрики 2.10.0
PyPI
patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
Власна ліцензія27 серп. 2026 р. · метрики 2.10.0
PyPI
galacticdynamics/phasecurvefit
Construct Paths through Phase-Space Points
MIT17 лип. 2026 р. · метрики 2.10.0
PyPI
google-deepmind/chex
Опис репозиторію не опубліковано.
Apache-2.027 серп. 2026 р. · метрики 2.10.0
PyPI
google-deepmind/dm-haiku
JAX-based neural network library
Apache-2.030 лип. 2026 р. · метрики 2.10.0
PyPI
mpi4jax/mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
MIT23 лип. 2026 р. · метрики 2.10.0
PyPI
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Apache-2.030 лип. 2026 р. · метрики 2.10.0
PyPI
patrick-kidger/lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
Apache-2.08 серп. 2026 р. · метрики 2.10.0
PyPI
patrick-kidger/optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Apache-2.08 серп. 2026 р. · метрики 2.10.0
PyPI
tensorflow/probability
Probabilistic reasoning and statistical analysis in TensorFlow
Apache-2.028 серп. 2026 р. · метрики 2.10.0
PyPI
michaelellis003/smcx
State-space inference in JAX: Kalman and particle filters, tempered SMC, and SMC²
Apache-2.026 лип. 2026 р. · метрики 2.10.0
PyPI
tillahoffmann/jax-mps
A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
Apache-2.018 лип. 2026 р. · метрики 2.10.0