Skip to content

Commit

Permalink
Update links to machine learning framework compatibility pages (#407)
Browse files Browse the repository at this point in the history
* update links to ml fw compatibility pages

* fix typo
  • Loading branch information
peterjunpark authored Jan 28, 2025
1 parent 58c3bae commit f1a6ea2
Show file tree
Hide file tree
Showing 3 changed files with 15 additions and 2 deletions.
6 changes: 6 additions & 0 deletions docs/install/3rd-party/jax-install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,12 @@ JAX on ROCm

This directory provides setup instructions and necessary files to build, test, and run JAX with ROCm support in a Docker environment, suitable for both runtime and CI workflows. Explore the following methods to use or build JAX on ROCm.

For hardware, software, and third-party framework compatibility between ROCm and JAX, see the following resources:

* :ref:`system-requirements`

* :doc:`rocm:compatibility/ml-compatibility/jax-compatibility`

Using a prebuilt Docker image
===========================================

Expand Down
5 changes: 3 additions & 2 deletions docs/install/3rd-party/pytorch-install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -27,10 +27,11 @@ To install PyTorch for ROCm, you have the following options:

<br/>

For hardware, software, and third-party framework compatibility between ROCm and PyTorch, refer to:
For hardware, software, and third-party framework compatibility between ROCm and PyTorch, see the following resources:

* :ref:`system-requirements`
* :ref:`3rd-party-support-matrix`

* :doc:`rocm:compatibility/ml-compatibility/pytorch-compatibility`

.. _using-docker-with-pytorch-pre-installed:

Expand Down
6 changes: 6 additions & 0 deletions docs/install/3rd-party/tensorflow-install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,12 @@ To install TensorFlow for ROCm, you have the following options:

* :ref:`install-tensorflow-wheels`

For hardware, software, and third-party framework compatibility between ROCm and TensorFlow, see the following resources:

* :ref:`system-requirements`

* :doc:`rocm:compatibility/ml-compatibility/tensorflow-compatibility`

.. note::

As of ROCm 6.1, ``tensorflow-rocm`` packages are found at `<https://repo.radeon.com/rocm/manylinux>`__.
Expand Down

0 comments on commit f1a6ea2

Please sign in to comment.