mirror of
https://github.com/ROCm/ROCm.git
synced 2026-01-09 06:38:00 -05:00
35 lines
1.2 KiB
ReStructuredText
35 lines
1.2 KiB
ReStructuredText
: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>`__
|