JAX MaxText training performance testing version history#
2025-06-20
2 min read time
Applies to Linux and Windows
This table lists previous versions of the ROCm JAX MaxText Docker image for training
performance testing. For detailed information about available models for
benchmarking, see the version-specific documentation.
You can find tagged
previous releases of the ROCm/jax-training Docker image on Docker Hub.
Image version |
ROCm version |
JAX version |
Resources |
|---|---|---|---|
25.5 |
6.3.4 |
0.4.35 |
|
25.4 |
6.3.0 |
0.4.31 |