Reverse Dependencies of orbax-checkpoint

The following projects have a declared dependency on orbax-checkpoint:

• apax — Atomistic Learned Potential Package in JAX
• brax — A differentiable physics engine written in JAX.
• chemtrain — Machine-learning deep molecular dynamics potentials.
• clip-jax — Training of CLIP in JAX
• cupbearer — A library for mechanistic anomaly detection
• deepmd-kit — A deep learning package for many-body potential energy representation and molecular dynamics
• e3md — Library for machine learning on physical tensors
• flax — Flax: A neural network library for JAX designed for flexibility
• flax-pilot — A Simplistic trainer for Flax
• gtech-optimus — Optimus library for real-time marketing personalization using RL.
• jax-ai-stack — no summary
• kauldron — Kauldron is a ML research library optimized for quick iteration and modularity.
• momaland — A standard API for Multi-Objective Multi-Agent Decision making and a diverse set of reference environments.
• nadl — Nasy's Deep Learning Toolkit
• neuroflex — An advanced neural network framework with interpretability, generalization, robustness, and fairness features
• orbax-export — Orbax Export
• pathwaysutils — Package of Pathways-on-Cloud utilities.
• paxml — Framework to configure and run machine learning experiments on top of Jax.
• penzai — Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
• reax — REAX: A simple training framework for JAX-based projects
• stars-omics — A spatial transcriptomics analysis tool.
• stories-jax — Learn spatially informed Waddington-like potentials for single-cell gene expression
• tax-dp — A fully sharded data parallel trainer for jax
• tractorun — Run distributed training in TractoAI