Reverse Dependencies of jmp

The following projects have a declared dependency on jmp:

• bax — A flexible trainer interface for Jax and Haiku.
• dm-haiku — Haiku is a library for building neural networks in JAX.
• haliax — Named Tensors for Legible Deep Learning in JAX
• lagrangebench — LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite
• levanter — Scalable Training for Foundation Models with Named Tensors and JAX
• neuroflex — An advanced neural network framework with interpretability, generalization, robustness, and fairness features
• pax3 — A stateful pytree library for training neural networks.
• progen-transformer — Protein Generation (ProGen)