# JAX-based ML research project
research jax distributed-training evaluation-framework
