Skip to content

Commit 279531f

Browse files
author
Bhanu Teja Goshikonda
committed
docs: club PyTorch and TensorFlow into a single training card
Merges the separate framework cards into one "Train Models" card covering PyTorch and TensorFlow, with links to both framework guides.
1 parent 8d458f9 commit 279531f

1 file changed

Lines changed: 2 additions & 10 deletions

File tree

docs/index.md

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -72,17 +72,9 @@ LLM serving is just one example. DLCs cover a range of AI/ML workloads — explo
7272

7373
---
7474

75-
Run distributed training with PyTorch on GPU or CPU, with EFA, NCCL, flash-attn, and DeepSpeed pre-installed.
75+
Run distributed training with PyTorch or TensorFlow on GPU or CPU, with EFA and NCCL pre-installed for multi-node workloads.
7676

77-
[PyTorch Guide](pytorch/index.md)
78-
79-
- **Train Models with TensorFlow**
80-
81-
---
82-
83-
Run TensorFlow training on Amazon SageMaker AI with EFA-capable multi-node support on Amazon Linux 2023.
84-
85-
[TensorFlow Guide](tensorflow/index.md)
77+
[PyTorch Guide](pytorch/index.md) · [TensorFlow Guide](tensorflow/index.md)
8678

8779
- **Build Your Own Image**
8880

0 commit comments

Comments
 (0)