Last released Feb 9, 2026
CLI tool for running JAX training on Google Cloud Spot TPUs with automatic preemption recovery and checkpoint resumption
Supported by