Beschreibung

JAX MaxText is a high-performance, open-source framework built on the Google JAX machine learning library to train LLMs at scale.

Pull Command
									    docker pull rocm/jax-training:maxtext-v25.4