Reverse Dependencies of orbax

The following projects have a declared dependency on orbax:

• bdax — Bayesian Data Analysis using PyMC and JAX
• chemise — Wrapper for training flax models
• chemtrain — Machine-learning deep molecular dynamics potentials.
• ciclo — no summary
• dalle-mini — DALL·E mini - Generate images from a text prompt
• felafax — felafax
• FlaxDiff — A versatile and easy to understand Diffusion library
• jax-ml-stack — no summary
• jaxagents — JAX implementation of Reinforcement Learning agents
• jaxlem — LEM slope stability analysis using JAX
• mdpax — GPU-accelerated MDP solvers in Python with JAX
• neuralgcm — Neural General Circulation Models for Weather and Climate
• scgpt — Large-scale generative pretrain of single cell using transformer.
• scPEFT — Harnessing the Power of Single Cell Large Language Models with Parameter Efficient Fine-Tuning.
• xminigrid — JAX-accelerated meta-reinforcement learning environments inspired by XLand and MiniGrid