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
96Exceptionalhealth index
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Python★ 534↓ 5.9M/moAug 29, 2026
Apache-2.0Aug 29, 2026 · metrics 2.10.0