mirror of
https://github.com/hanzoai/ROCm.git
synced 2026-11-02 19:03:06 +00:00
77 lines
2.9 KiB
ReStructuredText
77 lines
2.9 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
|
|
|
|
* - 26.2 (latest)
|
|
-
|
|
* ROCm 7.1.1
|
|
* JAX 0.8.2
|
|
-
|
|
* :doc:`Documentation <../jax-maxtext>`
|
|
* `Docker Hub <https://hub.docker.com/layers/rocm/jax-training/maxtext-v26.2/images/sha256-a89643388487b1e2fc6b6ef7bd3c44378c05d217309c977a1c18c72d05ebcaeb>`__
|
|
|
|
* - 26.1
|
|
-
|
|
* ROCm 7.1.1
|
|
* JAX 0.8.2
|
|
-
|
|
* :doc:`Documentation <jax-maxtext-v26.1>`
|
|
* `Docker Hub <https://hub.docker.com/layers/rocm/jax-training/maxtext-v26.1/images/sha256-901083bde353fe6362ada3036e452c792b2c96124e5900f4e9b5946c02ff9d6a>`__
|
|
|
|
* - 25.11
|
|
-
|
|
* ROCm 7.1.0
|
|
* JAX 0.7.1
|
|
-
|
|
* :doc:`Documentation <jax-maxtext-v25.11>`
|
|
* `Docker Hub <https://hub.docker.com/layers/rocm/jax-training/maxtext-v25.11/images/sha256-18e4d8f0b8ce7a7422c58046940dd5f32249960449fca09a562b65fb8eb1562a>`__
|
|
|
|
* - 25.9.1
|
|
-
|
|
* ROCm 7.0.0
|
|
* JAX 0.6.2
|
|
-
|
|
* :doc:`Documentation <jax-maxtext-v25.9>`
|
|
* `Docker Hub (25.9.1) <https://hub.docker.com/layers/rocm/jax-training/maxtext-v25.9.1/images/sha256-60946cfbd470f6ee361fc9da740233a4fb2e892727f01719145b1f7627a1cff6>`__
|
|
* `Docker Hub (25.9) <https://hub.docker.com/layers/rocm/jax-training/maxtext-v25.9/images/sha256-4bb16ab58279ef09cb7a5e362c38e3fe3f901de44d8dbac5d0cb3bac5686441e>`__
|
|
|
|
* - 25.7
|
|
-
|
|
* ROCm 6.4.1
|
|
* JAX 0.6.0, 0.5.0
|
|
-
|
|
* :doc:`Documentation <jax-maxtext-v25.7>`
|
|
* `Docker Hub (JAX 0.6.0) <https://hub.docker.com/layers/rocm/jax-training/maxtext-v25.7-jax060/images/sha256-7352212ae033a76dca2b9dceffc23c1b5f1a61a7a560082cf747a9bf1acfc9ce>`__
|
|
* `Docker Hub (JAX 0.5.0) <https://hub.docker.com/layers/rocm/jax-training/maxtext-v25.7/images/sha256-45f4c727d4019a63fc47313d3a5f5a5105569539294ddfd2d742218212ae9025>`__
|
|
|
|
* - 25.5
|
|
-
|
|
* ROCm 6.3.4
|
|
* JAX 0.4.35
|
|
-
|
|
* :doc:`Documentation <jax-maxtext-v25.5>`
|
|
* `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>`__
|