Усі теги
Тег каталогу

#jax

Усі репозиторії публічного реєстру з цим тегом — із тем GitHub або ключових слів, які публікують їхні реєстри пакетів. Здоров'я вимірюється за тією ж версіонованою методологією, що й решта реєстру.

68 записів
З тегом «jax»Упорядковано за індексом здоров'я
PyPI · Go · crates.io
100Винятковийіндекс здоров'я
wandb/wandb
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
Python · Go★ 11.2K↓ 26.6M/міс27 серп. 2026 р.
MIT27 серп. 2026 р. · метрики 2.10.0
PyPI
98Винятковийіндекс здоров'я
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.
Python★ 3 469↓ 199.1K/міс17 вер. 2026 р.
Apache-2.017 вер. 2026 р. · метрики 2.10.0
PyPI
98Винятковийіндекс здоров'я
keras-team/keras
Deep Learning for humans
Python★ 64.3K↓ 12.4M/міс20 вер. 2026 р.
Apache-2.020 вер. 2026 р. · метрики 2.10.0
PyPI
97Винятковийіндекс здоров'я
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Python★ 537↓ 2.8M/міс21 жовт. 2026 р.
Apache-2.021 жовт. 2026 р. · метрики 2.10.0
PyPI
97Винятковийіндекс здоров'я
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python★ 36.3K16 вер. 2026 р.
Apache-2.016 вер. 2026 р. · метрики 2.10.0
PyPI
96Винятковийіндекс здоров'я
iree-org/iree
A retargetable MLIR-based machine learning compiler and runtime toolkit.
C++ · C · MLIR★ 3 98620 жовт. 2026 р.
Apache-2.020 жовт. 2026 р. · метрики 2.10.0
PyPI
96Винятковийіндекс здоров'я
keras-team/keras-hub
Pretrained model hub for Keras 3.
Python★ 99521 жовт. 2026 р.
Apache-2.021 жовт. 2026 р. · метрики 2.10.0
PyPI
95Винятковийіндекс здоров'я
HIPS/autograd
Efficiently computes derivatives of NumPy code.
Python★ 7 523↓ 5.1M/міс28 серп. 2026 р.
MIT28 серп. 2026 р. · метрики 2.10.0
PyPI · crates.io
95Винятковийіндекс здоров'я
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
Jupyter Notebook · Python★ 7 304↓ 4.4M/міс27 серп. 2026 р.
Apache-2.027 серп. 2026 р. · метрики 2.10.0
PyPI
95Винятковийіндекс здоров'я
google/grain
Library for reading and processing ML training data.
Python★ 778↓ 797.2K/міс12 вер. 2026 р.
Apache-2.012 вер. 2026 р. · метрики 2.10.0
PyPI
94Винятковийіндекс здоров'я
pyro-ppl/numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Python★ 2 754↓ 812.2K/міс17 вер. 2026 р.
Apache-2.017 вер. 2026 р. · метрики 2.10.0
PyPI
92Відміннийіндекс здоров'я
tue-bmd/zea
zea: A Toolbox for Cognitive Ultrasound Imaging
Python★ 104↓ 2 713/міс1 жовт. 2026 р.
Apache-2.01 жовт. 2026 р. · метрики 2.10.0
PyPI
91Відміннийіндекс здоров'я
nstarman/quax
Multiple dispatch over abstract array types in JAX.
Python★ 143↓ 18.4K/міс4 жовт. 2026 р.
Apache-2.04 жовт. 2026 р. · метрики 2.10.0
PyPI
91Відміннийіндекс здоров'я
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Python★ 2 970↓ 1.1M/міс14 вер. 2026 р.
Apache-2.014 вер. 2026 р. · метрики 2.10.0
PyPI
90Відміннийіндекс здоров'я
GalacticDynamics/coordinax
Coordinates in JAX
Python★ 38↓ 4 706/міс16 жовт. 2026 р.
MIT16 жовт. 2026 р. · метрики 2.10.0
PyPI
90Відміннийіндекс здоров'я
GalacticDynamics/unxt
Unitful Quantities in JAX
Python★ 69↓ 14.1K/міс18 вер. 2026 р.
BSD-3-Clause18 вер. 2026 р. · метрики 2.10.0
PyPI
90Відміннийіндекс здоров'я
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.
Python★ 243↓ 175.1K/міс11 вер. 2026 р.
Apache-2.011 вер. 2026 р. · метрики 2.10.0
PyPI
89Відміннийіндекс здоров'я
GW-JAX-Team/ripple
Differentiable gravitational waveforms in JAX
Python★ 17↓ 4 223/міс14 жовт. 2026 р.
MIT14 жовт. 2026 р. · метрики 2.10.0
PyPI
89Відміннийіндекс здоров'я
instadeepai/jumanji
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
Python★ 865↓ 7 072/міс3 жовт. 2026 р.
Apache-2.03 жовт. 2026 р. · метрики 2.10.0
PyPI
89Відміннийіндекс здоров'я
thomaspinder/GPJax
Gaussian processes in JAX and Equinox.
Python★ 665↓ 8 632/міс19 вер. 2026 р.
MIT19 вер. 2026 р. · метрики 2.10.0
PyPI
88Відміннийіндекс здоров'я
google/etils
Collection of eclectic utils for python.
Python · Jupyter Notebook★ 259↓ 8.1M/міс27 серп. 2026 р.
Apache-2.027 серп. 2026 р. · метрики 2.10.0
PyPI
87Відміннийіндекс здоров'я
google-deepmind/dm-haiku
JAX-based neural network library
Python★ 3 28419 вер. 2026 р.
Apache-2.019 вер. 2026 р. · метрики 2.10.0
PyPI
86Відміннийіндекс здоров'я
arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Python · Jupyter Notebook★ 9 582↓ 27.1M/міс28 серп. 2026 р.
MIT28 серп. 2026 р. · метрики 2.10.0
PyPI
86Відміннийіндекс здоров'я
tensorflow/datasets
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Python★ 4 58128 серп. 2026 р.
Apache-2.028 серп. 2026 р. · метрики 2.10.0
PyPI
84Відміннийіндекс здоров'я
GW-JAX-Team/flowMC
Normalizing-flow enhanced sampling package for probabilistic inference in JAX
Python★ 7↓ 6 818/міс12 жовт. 2026 р.
MIT12 жовт. 2026 р. · метрики 2.10.0
PyPI
84Відміннийіндекс здоров'я
chaobrain/brainstate
A State-based Transformation System for Brain Modeling.
Python★ 22↓ 4 874/міс18 вер. 2026 р.
Apache-2.018 вер. 2026 р. · метрики 2.10.0
PyPI
84Відміннийіндекс здоров'я
exoplanet-dev/jaxoplanet
Astronomical time series analysis with JAX
Python★ 87↓ 2 689/міс19 жовт. 2026 р.
MIT19 жовт. 2026 р. · метрики 2.10.0
PyPI · npm
84Відміннийіндекс здоров'я
magenta/magenta-realtime
Magenta RealTime 2: An Open-Weights Live Music Model
Python · C++★ 1 791↓ 817/міс10 вер. 2026 р.
Apache-2.010 вер. 2026 р. · метрики 2.10.0
PyPI
83Відміннийіндекс здоров'я
GalacticDynamics/quax-blocks
Blocks for constructing `quax` classes
Python★ 3↓ 15K/міс20 жовт. 2026 р.
BSD-3-Clause20 жовт. 2026 р. · метрики 2.10.0
PyPI
83Відміннийіндекс здоров'я
GalacticDynamics/xmmutablemap
Immutable Map, compatible with Jax
Python★ 5↓ 15.4K/міс11 жовт. 2026 р.
MIT11 жовт. 2026 р. · метрики 2.10.0