Primus JAX MaxText training performance testing version history

Primus JAX MaxText training performance testing version history#

This table lists previous releases of the Primus JAX MaxText training Docker image. The following documentation archives use Primus with the MaxText backend and demonstrate training performance testing. You can find tagged previous releases of the ROCm/jax-training Docker image on Docker Hub.

Image version	Components	Resources
26.5 (latest)	ROCm 7.14.0 JAX 0.10.0	Documentation Docker Hub
26.4	ROCm 7.14.0 JAX 0.9.1	Documentation Docker Hub
26.3	ROCm 7.2.1 JAX 0.8.2	Documentation Docker Hub
26.2	ROCm 7.1.1 JAX 0.8.2	Documentation Docker Hub
26.1	ROCm 7.1.1 JAX 0.8.2	Documentation Docker Hub
25.11	ROCm 7.1.0 JAX 0.7.1	Documentation Docker Hub
25.9.1	ROCm 7.0.0 JAX 0.6.2	Documentation Docker Hub (25.9.1) Docker Hub (25.9)
25.7	ROCm 6.4.1 JAX 0.6.0, 0.5.0	Documentation Docker Hub (JAX 0.6.0) Docker Hub (JAX 0.5.0)
25.5	ROCm 6.3.4 JAX 0.4.35	Documentation Docker Hub
25.4	ROCm 6.3.0 JAX 0.4.31	Documentation Docker Hub