PyPI · crates.io · npm
Catstate101/superfermion
No repository description published.
Custom licenseOct 16, 2026 · metrics 2.10.0
All tags
Catalogue tag
#jax
Every repository in the public record carrying this tag — from its GitHub topics or the keywords its package registries publish. Health is measured under the same versioned methodology as the rest of the record.
68 records
PyPI
HighDimensionalEconLab/nlls_gram
Gram/dual-form Levenberg-Marquardt nonlinear least-squares solvers for JAX/Flax NNX models
MITSep 10, 2026 · metrics 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.
MITSep 9, 2026 · metrics 2.10.0
PyPI
OpenHCSDev/ArrayBridge
No repository description published.
MITOct 2, 2026 · metrics 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.
MITSep 9, 2026 · metrics 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.0Sep 10, 2026 · metrics 2.10.0
PyPI
42Ar/jaxon
Save pytrees efficiently in hdf5 files.
GPL-3.0Sep 17, 2026 · metrics 2.10.0
PyPI
google-deepmind/jraph
A Graph Neural Network Library in Jax
Apache-2.0Sep 17, 2026 · metrics 2.10.0