Last released Oct 16, 2024
Scaling utilities for JAX.
Last released Jun 24, 2024
TPU pod commander is a command line tool to manage cloud TPU pods.
Last released Jun 11, 2024
Machine learning experiment utils.
Last released Oct 3, 2018
Supported by
