:orphan:

********************************************************
JAX MaxText training performance testing version history
********************************************************

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 <https://hub.docker.com/r/rocm/jax-training/tags>`_.

.. list-table::
   :header-rows: 1

   * - Image version
     - Components
     - Resources

   * - 25.5 (latest)
     - 
       * ROCm 6.3.4
       * JAX 0.4.35
     - 
       * :doc:`Documentation <../jax-maxtext>`
       * `Docker Hub <https://hub.docker.com/layers/rocm/jax-training/maxtext-v25.5/images/sha256-4e0516358a227cae8f552fb866ec07e2edcf244756f02e7b40212abfbab5217b>`__

   * - 25.4
     - 
       * ROCm 6.3.0
       * JAX 0.4.31
     - 
       * :doc:`Documentation <jax-maxtext-v25.4>`
       * `Docker Hub <https://hub.docker.com/layers/rocm/jax-training/maxtext-v25.4/images/sha256-fb3eb71cd74298a7b3044b7130cf84113f14d518ff05a2cd625c11ea5f6a7b01>`__
