Compare commits

...

4 Commits

Author SHA1 Message Date
randyh62
8d183c2e95 Apply suggestion from @randyh62
OK
2026-01-15 15:53:46 -08:00
randyh62
3a7cfd3958 Apply suggestion from @randyh62
OK
2026-01-15 15:53:00 -08:00
randyh62
833fdf4c95 Apply suggestion from @randyh62
OK
2026-01-15 15:52:14 -08:00
randyh62
28f028d304 Update Deep Frameworks to use intersphinx links where possible 2026-01-15 15:48:31 -08:00
2 changed files with 60 additions and 83 deletions

View File

@@ -19,117 +19,104 @@ The table below summarizes information about ROCm-enabled deep learning framewor
:widths: 5 3 6 3 :widths: 5 3 6 3
* - Framework * - Framework
- Installation - Installation guide
- Installation options - Installation options
- GitHub - GitHub
* - `PyTorch <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/pytorch-compatibility.html>`__ * - :doc:`PyTorch <../compatibility/ml-compatibility/pytorch-compatibility>`
- .. raw:: html - :doc:`Pytorch install <rocm-install-on-linux:install/3rd-party/pytorch-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/pytorch-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/pytorch-install.html#using-a-docker-image-with-pytorch-pre-installed>`__ - Docker image
- `Wheels package <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/pytorch-install.html#using-a-wheels-package>`__ - Wheels package
- `ROCm Base Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/pytorch-install.html#using-the-pytorch-rocm-base-docker-image>`__ - ROCm Base Docker image
- `Upstream Docker file <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/pytorch-install.html#using-the-pytorch-upstream-dockerfile>`__ - Upstream Docker file
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/pytorch"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/pytorch"><i class="fab fa-github fa-lg"></i></a>
* - `TensorFlow <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/tensorflow-compatibility.html>`__ * - :doc:`TensorFlow <../compatibility/ml-compatibility/tensorflow-compatibility>`
- .. raw:: html - :doc:`TensorFlow install <rocm-install-on-linux:install/3rd-party/tensorflow-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/tensorflow-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/tensorflow-install.html#using-a-docker-image-with-tensorflow-pre-installed>`__ - Docker image
- `Wheels package <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/tensorflow-install.html#using-a-wheels-package>`__ - Wheels package
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/tensorflow-upstream"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/tensorflow-upstream"><i class="fab fa-github fa-lg"></i></a>
* - `JAX <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/jax-compatibility.html>`__ * - :doc:`JAX <../compatibility/ml-compatibility/jax-compatibility>`
- .. raw:: html - :doc:`JAX install <rocm-install-on-linux:install/3rd-party/jax-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/jax-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/jax-install.html#using-a-prebuilt-docker-image>`__ - Docker image
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/jax"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/jax"><i class="fab fa-github fa-lg"></i></a>
* - `verl <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/verl-compatibility.html>`__ * - :doc:`verl <../compatibility/ml-compatibility/verl-compatibility>`
- .. raw:: html - :doc:`verl install <rocm-install-on-linux:install/3rd-party/verl-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/verl-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/verl-install.html#use-a-prebuilt-docker-image-with-verl-pre-installed>`__ - Docker image
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/verl"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/verl"><i class="fab fa-github fa-lg"></i></a>
* - `Stanford Megatron-LM <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/stanford-megatron-lm-compatibility.html>`__ * - :doc:`Stanford Megatron-LM <../compatibility/ml-compatibility/stanford-megatron-lm-compatibility>`
- .. raw:: html - :doc:`Stanford Megatron-LM install <rocm-install-on-linux:install/3rd-party/stanford-megatron-lm-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/stanford-megatron-lm-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/stanford-megatron-lm-install.html#use-a-prebuilt-docker-image-with-stanford-megatron-lm-pre-installed>`__ - Docker image
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/Stanford-Megatron-LM"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/Stanford-Megatron-LM"><i class="fab fa-github fa-lg"></i></a>
* - `DGL <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/dgl-compatibility.html>`__ * - :doc:`DGL <../compatibility/ml-compatibility/dgl-compatibility>`
- .. raw:: html - :doc:`DGL install <rocm-install-on-linux:install/3rd-party/dgl-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/dgl-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/dgl-install.html#use-a-prebuilt-docker-image-with-dgl-pre-installed>`__ - Docker image
- `Wheels package <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/dgl-install.html#use-a-wheels-package>`__
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/dgl"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/dgl"><i class="fab fa-github fa-lg"></i></a>
* - `Megablocks <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/megablocks-compatibility.html>`__ * - :doc:`Megablocks <../compatibility/ml-compatibility/megablocks-compatibility>`
- .. raw:: html - :doc:`Megablocks install <rocm-install-on-linux:install/3rd-party/megablocks-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/megablocks-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/megablocks-install.html#using-a-prebuilt-docker-image-with-megablocks-pre-installed>`__ - Docker image
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/megablocks"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/megablocks"><i class="fab fa-github fa-lg"></i></a>
* - `Ray <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/ray-compatibility.html>`__ * - :doc:`Taichi <../compatibility/ml-compatibility/taichi-compatibility>`
- `Taichi install <https://rocm.docs.amd.com/projects/taichi/en/latest/install/taichi-install.html>`__
-
- Docker image
- Wheels package
- .. raw:: html - .. raw:: html
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/ray-install.html"><i class="fas fa-link fa-lg"></i></a> <a href="https://github.com/ROCm/taichi"><i class="fab fa-github fa-lg"></i></a>
* - :doc:`Ray <../compatibility/ml-compatibility/ray-compatibility>`
- :doc:`Ray install <rocm-install-on-linux:install/3rd-party/ray-install>`
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/ray-install.html#using-a-prebuilt-docker-image-with-ray-pre-installed>`__ - Docker image
- `Wheels package <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/ray-install.html#install-ray-on-bare-metal-or-a-custom-container>`__ - Wheels package
- `ROCm Base Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/ray-install.html#build-your-own-docker-image>`__ - ROCm Base Docker image
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/ray"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/ray"><i class="fab fa-github fa-lg"></i></a>
* - `llama.cpp <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/llama-cpp-compatibility.html>`__ * - :doc:`llama.cpp <../compatibility/ml-compatibility/llama-cpp-compatibility>`
- .. raw:: html - :doc:`llama.cpp install <rocm-install-on-linux:install/3rd-party/llama-cpp-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/llama-cpp-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/llama-cpp-install.html#use-a-prebuilt-docker-image-with-llama-cpp-pre-installed>`__ - Docker image
- `ROCm Base Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/llama-cpp-install.html#build-your-own-docker-image>`__ - ROCm Base Docker image
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/llama.cpp"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/llama.cpp"><i class="fab fa-github fa-lg"></i></a>
* - `FlashInfer <https://rocm.docs.amd.com/en/latest/compatibility/ml-compatibility/flashinfer-compatibility.html>`__ * - :doc:`FlashInfer <../compatibility/ml-compatibility/flashinfer-compatibility>`
- .. raw:: html - :doc:`FlashInfer install <rocm-install-on-linux:install/3rd-party/flashinfer-install>`
<a href="https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/flashinfer-install.html"><i class="fas fa-link fa-lg"></i></a>
- -
- `Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/flashinfer-install.html#use-a-prebuilt-docker-image-with-flashinfer-pre-installed>`__ - Docker image
- `ROCm Base Docker image <https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/flashinfer-install.html#build-your-own-docker-image>`__ - ROCm Base Docker image
- .. raw:: html - .. raw:: html
<a href="https://github.com/ROCm/flashinfer"><i class="fab fa-github fa-lg"></i></a> <a href="https://github.com/ROCm/flashinfer"><i class="fab fa-github fa-lg"></i></a>

View File

@@ -25,29 +25,31 @@ subtrees:
title: HIP SDK on Windows title: HIP SDK on Windows
- url: https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/index.html - url: https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/index.html
title: ROCm on Radeon and Ryzen title: ROCm on Radeon and Ryzen
- file: how-to/deep-learning-rocm.md - file: how-to/deep-learning-rocm
title: Deep learning frameworks title: Deep learning frameworks
subtrees: subtrees:
- entries: - entries:
- file: compatibility/ml-compatibility/pytorch-compatibility.rst - file: compatibility/ml-compatibility/pytorch-compatibility
title: PyTorch compatibility title: PyTorch compatibility
- file: compatibility/ml-compatibility/tensorflow-compatibility.rst - file: compatibility/ml-compatibility/tensorflow-compatibility
title: TensorFlow compatibility title: TensorFlow compatibility
- file: compatibility/ml-compatibility/jax-compatibility.rst - file: compatibility/ml-compatibility/jax-compatibility
title: JAX compatibility title: JAX compatibility
- file: compatibility/ml-compatibility/verl-compatibility.rst - file: compatibility/ml-compatibility/verl-compatibility
title: verl compatibility title: verl compatibility
- file: compatibility/ml-compatibility/stanford-megatron-lm-compatibility.rst - file: compatibility/ml-compatibility/stanford-megatron-lm-compatibility
title: Stanford Megatron-LM compatibility title: Stanford Megatron-LM compatibility
- file: compatibility/ml-compatibility/dgl-compatibility.rst - file: compatibility/ml-compatibility/dgl-compatibility
title: DGL compatibility title: DGL compatibility
- file: compatibility/ml-compatibility/megablocks-compatibility.rst - file: compatibility/ml-compatibility/megablocks-compatibility
title: Megablocks compatibility title: Megablocks compatibility
- file: compatibility/ml-compatibility/ray-compatibility.rst - file: compatibility/ml-compatibility/taichi-compatibility
title: Taichi compatibility
- file: compatibility/ml-compatibility/ray-compatibility
title: Ray compatibility title: Ray compatibility
- file: compatibility/ml-compatibility/llama-cpp-compatibility.rst - file: compatibility/ml-compatibility/llama-cpp-compatibility
title: llama.cpp compatibility title: llama.cpp compatibility
- file: compatibility/ml-compatibility/flashinfer-compatibility.rst - file: compatibility/ml-compatibility/flashinfer-compatibility
title: FlashInfer compatibility title: FlashInfer compatibility
- file: how-to/build-rocm.rst - file: how-to/build-rocm.rst
title: Build ROCm from source title: Build ROCm from source
@@ -75,14 +77,8 @@ subtrees:
- entries: - entries:
- file: how-to/rocm-for-ai/training/benchmark-docker/primus-megatron.rst - file: how-to/rocm-for-ai/training/benchmark-docker/primus-megatron.rst
title: Train a model with Primus and Megatron-LM title: Train a model with Primus and Megatron-LM
entries:
- file: how-to/rocm-for-ai/training/benchmark-docker/megatron-lm.rst
title: Train a model with Megatron-LM
- file: how-to/rocm-for-ai/training/benchmark-docker/primus-pytorch.rst - file: how-to/rocm-for-ai/training/benchmark-docker/primus-pytorch.rst
title: Train a model with Primus and PyTorch title: Train a model with Primus and PyTorch
entries:
- file: how-to/rocm-for-ai/training/benchmark-docker/pytorch-training.rst
title: Train a model with PyTorch
- file: how-to/rocm-for-ai/training/benchmark-docker/jax-maxtext.rst - file: how-to/rocm-for-ai/training/benchmark-docker/jax-maxtext.rst
title: Train a model with JAX MaxText title: Train a model with JAX MaxText
- file: how-to/rocm-for-ai/training/benchmark-docker/mpt-llm-foundry - file: how-to/rocm-for-ai/training/benchmark-docker/mpt-llm-foundry
@@ -121,8 +117,6 @@ subtrees:
title: SGLang inference performance testing title: SGLang inference performance testing
- file: how-to/rocm-for-ai/inference/benchmark-docker/sglang-distributed.rst - file: how-to/rocm-for-ai/inference/benchmark-docker/sglang-distributed.rst
title: SGLang distributed inference with Mooncake title: SGLang distributed inference with Mooncake
- file: how-to/rocm-for-ai/inference/xdit-diffusion-inference.rst
title: xDiT diffusion inference
- file: how-to/rocm-for-ai/inference/deploy-your-model.rst - file: how-to/rocm-for-ai/inference/deploy-your-model.rst
title: Deploy your model title: Deploy your model
@@ -140,8 +134,6 @@ subtrees:
title: Profile and debug title: Profile and debug
- file: how-to/rocm-for-ai/inference-optimization/workload.rst - file: how-to/rocm-for-ai/inference-optimization/workload.rst
title: Workload optimization title: Workload optimization
- file: how-to/rocm-for-ai/inference-optimization/vllm-optimization.rst
title: vLLM V1 performance optimization
- url: https://rocm.docs.amd.com/projects/ai-developer-hub/en/latest/ - url: https://rocm.docs.amd.com/projects/ai-developer-hub/en/latest/
title: AI tutorials title: AI tutorials
@@ -188,7 +180,7 @@ subtrees:
- file: conceptual/gpu-arch/mi300-mi200-performance-counters.rst - file: conceptual/gpu-arch/mi300-mi200-performance-counters.rst
title: MI300 and MI200 performance counters title: MI300 and MI200 performance counters
- file: conceptual/gpu-arch/mi350-performance-counters.rst - file: conceptual/gpu-arch/mi350-performance-counters.rst
title: MI350 Series performance counters title: MI350 series performance counters
- file: conceptual/gpu-arch/mi250.md - file: conceptual/gpu-arch/mi250.md
title: MI250 microarchitecture title: MI250 microarchitecture
subtrees: subtrees:
@@ -222,8 +214,6 @@ subtrees:
title: ROCm tools, compilers, and runtimes title: ROCm tools, compilers, and runtimes
- file: reference/gpu-arch-specs.rst - file: reference/gpu-arch-specs.rst
- file: reference/gpu-atomics-operation.rst - file: reference/gpu-atomics-operation.rst
- file: reference/env-variables.rst
title: Environment variables
- file: reference/precision-support.rst - file: reference/precision-support.rst
title: Data types and precision support title: Data types and precision support
- file: reference/graph-safe-support.rst - file: reference/graph-safe-support.rst