All tags
Catalogue tag

#jax-machine-learning

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.

1 record
Tagged “jax-machine-learning”Ranked by health index
PyPI
97Exceptionalhealth index
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Python★ 537↓ 2.8M/moOct 21, 2026
Apache-2.0Oct 21, 2026 · metrics 2.10.0