Тег каталогу

#jax

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

46 записів

З тегом «jax»Упорядковано за індексом здоров'я

PyPI

75Добрийіндекс здоров'я

avitai/artifex

A research-focused modular generative modeling library built on JAX/Flax NNX

Python★ 1↓ 2 893/міс29 серп. 2026 р.

MIT29 серп. 2026 р. · метрики 2.10.0

PyPI

75Добрийіндекс здоров'я

tensorcircuit/tensorcircuit-ng

Next-gen AI-native tensor-network-based quantum software framework

Python★ 87↓ 2 149/міс22 серп. 2026 р.

Apache-2.022 серп. 2026 р. · метрики 2.10.0

PyPI

73Добрийіндекс здоров'я

google-deepmind/distrax

Опис репозиторію не опубліковано.

Python★ 645↓ 175.1K/міс30 лип. 2026 р.

Apache-2.030 лип. 2026 р. · метрики 2.10.0

PyPI

71Добрийіндекс здоров'я

araffin/sbx

SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms

Python★ 608↓ 1 960/міс6 вер. 2026 р.

MIT6 вер. 2026 р. · метрики 2.10.0

PyPI · npm

71Добрийіндекс здоров'я

explosion/thinc

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

Python★ 2 890↓ 25.7M/міс27 серп. 2026 р.

MIT27 серп. 2026 р. · метрики 2.10.0

PyPI

71Добрийіндекс здоров'я

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 р. · метрики 2.10.0

PyPI

69Добрийіндекс здоров'я

google/jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

Python★ 1 0554 серп. 2026 р.

Apache-2.04 серп. 2026 р. · метрики 2.10.0

PyPI

63Помірнийіндекс здоров'я

OpenHCSDev/PolyStore

Опис репозиторію не опубліковано.

Python★ 0↓ 9 937/міс22 серп. 2026 р.

MIT22 серп. 2026 р. · метрики 2.10.0

PyPI

63Помірнийіндекс здоров'я

elkins-lab/diff-biophys

Differentiable biophysical modeling in JAX (SAXS, NMR, CD), with Jupyter Notebook examples.

Python★ 017 лип. 2026 р.

MIT17 лип. 2026 р. · метрики 2.10.0

PyPI

59Помірнийіндекс здоров'я

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 р. · метрики 2.10.0

PyPI

59Помірнийіндекс здоров'я

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 р. · метрики 2.10.0

PyPI

57Помірнийіндекс здоров'я

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 р. · метрики 2.10.0

PyPI

57Помірнийіндекс здоров'я

elkins-lab/diff-ensemble

Differentiable VAE framework for predicting protein structural ensembles (IDPs) consistent with SAXS and NMR data. Built on JAX/Flax.