AI-Hypercomputer / maxtext

A simple, performant and scalable Jax LLM!
Apache License 2.0
1.47k stars 275 forks source link

Report hyperparamters from the distributed training benchmark workload #828

Closed bernardhan33 closed 1 month ago

bernardhan33 commented 1 month ago

See cl/663597617 for the motivation of this change and the tests done.