Skip to content

Commit d34280a

Browse files
authored
Merge pull request #193 from NYU-RTS/pytorch_ddp
Pytorch ddp
2 parents bc2c3ac + f296282 commit d34280a

8 files changed

Lines changed: 796 additions & 6 deletions

File tree

docs/genai/01_getting_started/01_intro.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ If you're looking to harness Generative AI for administrative or classroom use,
66

77
Welcome to Pythia, the generative AI platform for research workflows. As part of the Pythia platform, the following capabilities are offered:
88
- [Access to externally hosted LLMs](../01_getting_started/02_llm_gateway.mdx)
9-
- [HPC resources for fine tuning LLMs](../../hpc/08_ml_ai_hpc/04_llm_fine_tuning.md)
9+
- [HPC resources for fine tuning LLMs](../../hpc/08_ml_ai_hpc/05_llm_fine_tuning.md)
1010

1111
:::info[Personal use]
1212
If you want to access NYU provided LLMs for personal use, proceed to https://gemini.google.com/app with your NYU credentials.
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
# Fine tuning
22

3-
Please look into harnessing RAG before attempting to fine-tune a model. For open-weight models, you can use the HPC cluster to perform LoRA fine-tuning as [described here](../../hpc/08_ml_ai_hpc/04_llm_fine_tuning.md).
3+
Please look into harnessing RAG before attempting to fine-tune a model. For open-weight models, you can use the HPC cluster to perform LoRA fine-tuning as [described here](../../hpc/08_ml_ai_hpc/05_llm_fine_tuning.md).

docs/hpc/08_ml_ai_hpc/01_intro.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,7 @@
1-
# Machine Learning on HPC
1+
# Machine Learning (ML) and Artificial Intelligence (AI) on HPC
2+
3+
Many people are interested in running ML and/or AI workflows on the HPC resources. To help facilitate this we have created this section that will provide examples of how one might use our resources for ML and AI tasks.
4+
5+
For ML we'll cover two prominent open-source deep learning frameworks, PyTorch and TensorFlow (coming soon). We'll show how to start with a single GPU example and then show more sophisticated examples.
6+
7+
For AI we'll cover how to run a Hugging Face Large Language Model (LLM) and how to fine tune an LLM.

docs/hpc/08_ml_ai_hpc/02_pytorch_intro.md

Lines changed: 353 additions & 0 deletions
Large diffs are not rendered by default.

docs/hpc/08_ml_ai_hpc/02_pytorch_on_hpc.md

Lines changed: 0 additions & 3 deletions
This file was deleted.

docs/hpc/08_ml_ai_hpc/03_pytorch_dpp.md

Lines changed: 434 additions & 0 deletions
Large diffs are not rendered by default.
File renamed without changes.

0 commit comments

Comments
 (0)