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

#jax

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

24 записи
З тегом «jax»Упорядковано за індексом здоров'я
PyPI · Go · crates.io
94Відміннийіндекс здоров'я
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↓ 23M/міс20 лип. 2026 р.
MIT20 лип. 2026 р. · метрики 1.13.0
PyPI
86Відміннийіндекс здоров'я
keras-team/keras
Deep Learning for humans
Python★ 64.2K↓ 18.4M/міс16 лип. 2026 р.
Apache-2.016 лип. 2026 р. · метрики 1.13.0
PyPI
83Добрийіндекс здоров'я
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python★ 36K16 лип. 2026 р.
Apache-2.016 лип. 2026 р. · метрики 1.13.0
PyPI
82Добрийіндекс здоров'я
HIPS/autograd
Efficiently computes derivatives of NumPy code.
Python★ 7 51218 лип. 2026 р.
MIT18 лип. 2026 р. · метрики 1.13.0
PyPI
81Добрийіндекс здоров'я
google/grain
Library for reading and processing ML training data.
Python★ 75221 лип. 2026 р.
Apache-2.021 лип. 2026 р. · метрики 1.13.0
PyPI · crates.io
78Добрийіндекс здоров'я
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
Jupyter Notebook · Python★ 7 271↓ 4.3M/міс18 лип. 2026 р.
Apache-2.018 лип. 2026 р. · метрики 1.13.0
PyPI
75Добрийіндекс здоров'я
tensorflow/datasets
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Python★ 4 58221 лип. 2026 р.
Apache-2.021 лип. 2026 р. · метрики 1.13.0
PyPI
74Добрийіндекс здоров'я
google/etils
Collection of eclectic utils for python.
Python · Jupyter Notebook★ 257↓ 8.3M/міс18 лип. 2026 р.
Apache-2.018 лип. 2026 р. · метрики 1.13.0
PyPI
72Добрийіндекс здоров'я
arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Python · Jupyter Notebook★ 9 54817 лип. 2026 р.
MIT17 лип. 2026 р. · метрики 1.13.0
PyPI · npm
72Добрийіндекс здоров'я
magenta/magenta-realtime
Magenta RealTime 2: An Open-Weights Live Music Model
Python · C++★ 1 683↓ 817/міс18 лип. 2026 р.
Apache-2.018 лип. 2026 р. · метрики 1.13.0
PyPI
71Добрийіндекс здоров'я
google-deepmind/chex
Опис репозиторію не опубліковано.
Python★ 948↓ 2M/міс21 лип. 2026 р.
Apache-2.021 лип. 2026 р. · метрики 1.13.0
PyPI
71Добрийіндекс здоров'я
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★ 231↓ 116.4K/міс19 лип. 2026 р.
Apache-2.019 лип. 2026 р. · метрики 1.13.0
PyPI
71Добрийіндекс здоров'я
patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
Python★ 1 842↓ 6.4M/міс18 лип. 2026 р.
Власна ліцензія18 лип. 2026 р. · метрики 1.13.0
PyPI
68Помірнийіндекс здоров'я
galacticdynamics/phasecurvefit
Construct Paths through Phase-Space Points
Python★ 417 лип. 2026 р.
MIT17 лип. 2026 р. · метрики 1.13.0
PyPI
65Помірнийіндекс здоров'я
tillahoffmann/jax-mps
A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
C++ · Python★ 192↓ 3 492/міс18 лип. 2026 р.
Apache-2.018 лип. 2026 р. · метрики 1.13.0
PyPI
62Помірнийіндекс здоров'я
tenax-lab/tenax
JAX-based tensor network library with label-based contraction and symmetry-aware block-sparse tensors
Python★ 719 лип. 2026 р.
Apache-2.019 лип. 2026 р. · метрики 1.13.0
PyPI
62Помірнийіндекс здоров'я
tensorcircuit/tensorcircuit-ng
Next-gen AI-native tensor-network-based quantum software framework
Python★ 81↓ 0/міс13 лип. 2026 р.
Apache-2.013 лип. 2026 р. · метрики 1.13.0
PyPI · npm
61Помірнийіндекс здоров'я
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
Python★ 2 890↓ 22.6M/міс18 лип. 2026 р.
MIT18 лип. 2026 р. · метрики 1.13.0
PyPI
60Помірнийіндекс здоров'я
elkins-lab/diff-biophys
Differentiable biophysical modeling in JAX (SAXS, NMR, CD), with Jupyter Notebook examples.
Python★ 017 лип. 2026 р.
MIT17 лип. 2026 р. · метрики 1.13.0
PyPI
56Помірнийіндекс здоров'я
elkins-lab/diff-ensemble
Differentiable VAE framework for predicting protein structural ensembles (IDPs) consistent with SAXS and NMR data. Built on JAX/Flax.
Python★ 1↓ 158/міс17 лип. 2026 р.
MIT17 лип. 2026 р. · метрики 1.13.0
PyPI
56Помірнийіндекс здоров'я
elkins-lab/resonance-flow
JAX-native differentiable protein folding framework integrating experimental NMR constraints (RDCs) and biophysical "self-correction." Several Jupyter Notebooks visualize the concepts.
Python★ 117 лип. 2026 р.
MIT17 лип. 2026 р. · метрики 1.13.0
PyPI
55Помірнийіндекс здоров'я
HighDimensionalEconLab/nlls_gram
Gram/dual-form Levenberg-Marquardt nonlinear least-squares solvers for JAX/Flax NNX models
Python★ 1↓ 2 882/міс18 лип. 2026 р.
MIT18 лип. 2026 р. · метрики 1.13.0
PyPI
55Помірнийіндекс здоров'я
ai-hypercomputer/jetstream
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
Python★ 45118 лип. 2026 р.
Apache-2.018 лип. 2026 р. · метрики 1.13.0
PyPI
55Помірнийіндекс здоров'я
google/jetstream
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
Python★ 45118 лип. 2026 р.
Apache-2.018 лип. 2026 р. · метрики 1.13.0