PyPI
avitai/artifex
A research-focused modular generative modeling library built on JAX/Flax NNX
MIT29. Aug. 2026 · Metriken 2.10.0
Alle Tags
Katalog-Tag
#jax
Alle Repositories im öffentlichen Register, die dieses Tag tragen — aus ihren GitHub-Topics oder den von ihren Paket-Registries veröffentlichten Schlagwörtern. Die Gesundheit wird nach derselben versionierten Methodik gemessen wie im übrigen Register.
46 Einträge
PyPI
tensorcircuit/tensorcircuit-ng
Next-gen AI-native tensor-network-based quantum software framework
Apache-2.022. Aug. 2026 · Metriken 2.10.0
PyPI
google-deepmind/distrax
Keine Repository-Beschreibung veröffentlicht.
Apache-2.030. Juli 2026 · Metriken 2.10.0
PyPI
araffin/sbx
SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
MIT6. Sept. 2026 · Metriken 2.10.0
PyPI · npm
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
MIT27. Aug. 2026 · Metriken 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. Juli 2026 · Metriken 2.10.0
PyPI
google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
Apache-2.04. Aug. 2026 · Metriken 2.10.0
PyPI
OpenHCSDev/PolyStore
Keine Repository-Beschreibung veröffentlicht.
MIT22. Aug. 2026 · Metriken 2.10.0
PyPI
elkins-lab/diff-biophys
Differentiable biophysical modeling in JAX (SAXS, NMR, CD), with Jupyter Notebook examples.
MIT17. Juli 2026 · Metriken 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. Juli 2026 · Metriken 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. Juli 2026 · Metriken 2.10.0
PyPI
HighDimensionalEconLab/nlls_gram
Gram/dual-form Levenberg-Marquardt nonlinear least-squares solvers for JAX/Flax NNX models
MIT18. Juli 2026 · Metriken 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. Juli 2026 · Metriken 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. Juli 2026 · Metriken 2.10.0
PyPI
42Ar/jaxon
Save pytrees efficiently in hdf5 files.
GPL-3.011. Aug. 2026 · Metriken 2.10.0
PyPI
google-deepmind/jraph
A Graph Neural Network Library in Jax
Apache-2.013. Aug. 2026 · Metriken 2.10.0