Rare find

jax_trainer. Lightning-like training API for JAX with Flax

github.com/phlippe/jax_trainer
by @phlippe

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.