phlippe/jax_trainer - Gitstar Ranking

Gitstar Ranking

Users
Organizations
Repositories
Rankings

Sign in with GitHub

phlippe

Fetched on 2026/03/14 10:03

phlippe / jax_trainer

Lightning-like training API for JAX with Flax - View it on GitHub

Star

45

Rank

559313

Released by @k0kubun in December 2014. Fork me on GitHub.