phlippe / jax_trainerLinks

Lightning-like training API for JAX with Flax
41Updated 6 months ago

Alternatives and similar repositories for jax_trainer

Users that are interested in jax_trainer are comparing it to the libraries listed below

Sorting: