PyPI
avitai/artifex
A research-focused modular generative modeling library built on JAX/Flax NNX
MIT29 ago 2026 · métricas 2.10.0
Todas las etiquetas
Etiqueta del catálogo
#jax
Todos los repositorios del registro público que llevan esta etiqueta, procedente de sus topics de GitHub o de las palabras clave que publican sus registros de paquetes. La salud se mide con la misma metodología versionada que el resto del registro.
46 registros
PyPI
tensorcircuit/tensorcircuit-ng
Next-gen AI-native tensor-network-based quantum software framework
Apache-2.022 ago 2026 · métricas 2.10.0
PyPI
google-deepmind/distrax
El repositorio no publica descripción.
Apache-2.030 jul 2026 · métricas 2.10.0
PyPI
araffin/sbx
SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
MIT6 sept 2026 · métricas 2.10.0
PyPI · npm
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
MIT27 ago 2026 · métricas 2.10.0
PyPI
tenax-lab/tenax
JAX-based tensor network library with label-based contraction and symmetry-aware block-sparse tensors
Apache-2.019 jul 2026 · métricas 2.10.0
PyPI
google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
Apache-2.04 ago 2026 · métricas 2.10.0
PyPI
OpenHCSDev/PolyStore
El repositorio no publica descripción.
MIT22 ago 2026 · métricas 2.10.0
PyPI
elkins-lab/diff-biophys
Differentiable biophysical modeling in JAX (SAXS, NMR, CD), with Jupyter Notebook examples.
MIT17 jul 2026 · métricas 2.10.0
PyPI
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).
Apache-2.018 jul 2026 · métricas 2.10.0
PyPI
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.
MIT17 jul 2026 · métricas 2.10.0
PyPI
HighDimensionalEconLab/nlls_gram
Gram/dual-form Levenberg-Marquardt nonlinear least-squares solvers for JAX/Flax NNX models
MIT18 jul 2026 · métricas 2.10.0
PyPI
elkins-lab/diff-ensemble
Differentiable VAE framework for predicting protein structural ensembles (IDPs) consistent with SAXS and NMR data. Built on JAX/Flax.
MIT17 jul 2026 · métricas 2.10.0
PyPI
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).
Apache-2.018 jul 2026 · métricas 2.10.0
PyPI
42Ar/jaxon
Save pytrees efficiently in hdf5 files.
GPL-3.011 ago 2026 · métricas 2.10.0
PyPI
google-deepmind/jraph
A Graph Neural Network Library in Jax
Apache-2.013 ago 2026 · métricas 2.10.0