argonne-lcf/ATPESC_MachineLearning

Lecture and hands-on material for Track 6 - AI/ML - of Argonne Training Program on Extreme-Scale Computing

Python

51

433 commits

updated Aug 7, 2026

See the code

README

ATPESC 2026

Agenda

Monday, August 3

TimeTalkSpeaker
8:30AMWelcome and IntroductionShilpika, ANL
8:35AMTransition time: splitting into groups (people new to deep learning vs. more experienced)
8:35AMParallel Session
- Main room: Introduction to Deep LearningHuihuo Zheng, ANL
- Breakout room: Distributed Deep LearningBethany Lusch, ANL
9:50AMBreak
10:15AMProfiling Deep LearningNathan Nichols, ANL
11:20 AMIntroduction to Large Language Models (LLMs)Jingyan (Jane) Jiang, ANL
12:00PMPre-training on a Supercomputer - Part 1Sam Foreman, ANL
12:30PMLunch
1:30PMPre-training on a Supercomputer - Part 2Sam Foreman, ANL
2:30PMPost-training (Finetuning/Alignment/RL techniques)Filippo Simini, ANL
4:00PMExplainable AI for ScienceShilpika, ANL
4:30PMBreak
5:30PMFeatured SpeakerJack Dongarra

Tuesday, August 4

TimeTalkSpeaker
8:30AMWelcome and introductionShilpika, ANL
8:35AMWorkflow Tools for ScienceChristine Simpson, ANL
9:30AMCoupled Workflows for Science (Simulations + AI Workflows)Riccardo Balin and Christine Simpson, ANL
10:30AMBreak
10:50AMInferenceMisha Salim, ANL
11:50 AMAgentic Tools - Part 1 (Trinity/Hermes/OpenClaw)Huihuo Zheng, ANL
12:30 PMLunch
1:30 PMAI TestbedVaruni Sastry and Murali Emani, ANL
3:00PMAgentic Workflows for ScienceThang Pham, ANL
4:00PMAgentic Tools - Part 2 (Academy)Kyle Chard, UChicago/ANL
4:30PMBreak
5:30PMFeatured SpeakerBill Gropp

At the beginning of the first day, we will temporarily split into two groups. Attendees can choose between Introduction to Deep Learning and Distributed Deep Learning.

The "Introduction to deep learning" session will rely on Jupyter Notebooks which are targeted for running on Google's Colaboratory Platform or ALCF JupyterHub. The Colab platform gives the user a virtual machine in which to run Python codes including machine learning codes. The VM comes with a preinstalled environment that includes most of what is needed for these tutorials.

The other sessions involve Python scripts executed on the Aurora and AI Testbed platforms at ALCF.

Reservations

  • Queue:
    • Daytime reservations: -q ATPESC
    • Evening reservations: -q ATPESC-Night
    • Outside of reservations: -q debug or -q prod (more info)
  • Project/Allocation: ATPESC2026 (-A ATPESC2026)
  • Shared directories:
    • Aurora: /flare/ATPESC2026
    • Polaris: /eagle/ATPESC2026
  • ALCF Unix Groups: ATPESC2026

Using Google Colab

Google Colab involves running Jupyter notebooks.

Do the following before you come to the tutorial:

  • You need a Google Account to use Colaboratory
  • Go to Google's Colaboratory Platform
  • You should see this page start_page
  • Now you can open the File menu at the top left and select Open Notebook which will open a dialogue box.
  • Select the GitHub tab in the dialogue box.
  • From here you can enter the url for the github repo: https://github.com/argonne-lcf/ATPESC_MachineLearning and hit <enter>. open_github
  • This will show you a list of the Notebooks available in the repo. When you select a notebook from this list it will create a copy for you in your Colaboratory account (all *.ipynb files in the Colaboratory account will be stored in your Google Drive).
  • To use a GPU in the notbook select Runtime -> Change Runtime Type and select an accelerator.

Weights & Biases API key

For the Pre-training on a Supercomputer session, you will need a Wandb api_key. Visit https://docs.wandb.ai/quickstart/ to sign-up and get the key.

argonne-lcf/ATPESC_MachineLearning

Lecture and hands-on material for Track 6 - AI/ML - of Argonne Training Program on Extreme-Scale Computing

Python

51

433 commits

updated Aug 7, 2026

See the code

README

ATPESC 2026

Agenda

Monday, August 3

TimeTalkSpeaker
8:30AMWelcome and IntroductionShilpika, ANL
8:35AMTransition time: splitting into groups (people new to deep learning vs. more experienced)
8:35AMParallel Session
- Main room: Introduction to Deep LearningHuihuo Zheng, ANL
- Breakout room: Distributed Deep LearningBethany Lusch, ANL
9:50AMBreak
10:15AMProfiling Deep LearningNathan Nichols, ANL
11:20 AMIntroduction to Large Language Models (LLMs)Jingyan (Jane) Jiang, ANL
12:00PMPre-training on a Supercomputer - Part 1Sam Foreman, ANL
12:30PMLunch
1:30PMPre-training on a Supercomputer - Part 2Sam Foreman, ANL
2:30PMPost-training (Finetuning/Alignment/RL techniques)Filippo Simini, ANL
4:00PMExplainable AI for ScienceShilpika, ANL
4:30PMBreak
5:30PMFeatured SpeakerJack Dongarra

Tuesday, August 4

TimeTalkSpeaker
8:30AMWelcome and introductionShilpika, ANL
8:35AMWorkflow Tools for ScienceChristine Simpson, ANL
9:30AMCoupled Workflows for Science (Simulations + AI Workflows)Riccardo Balin and Christine Simpson, ANL
10:30AMBreak
10:50AMInferenceMisha Salim, ANL
11:50 AMAgentic Tools - Part 1 (Trinity/Hermes/OpenClaw)Huihuo Zheng, ANL
12:30 PMLunch
1:30 PMAI TestbedVaruni Sastry and Murali Emani, ANL
3:00PMAgentic Workflows for ScienceThang Pham, ANL
4:00PMAgentic Tools - Part 2 (Academy)Kyle Chard, UChicago/ANL
4:30PMBreak
5:30PMFeatured SpeakerBill Gropp

At the beginning of the first day, we will temporarily split into two groups. Attendees can choose between Introduction to Deep Learning and Distributed Deep Learning.

The "Introduction to deep learning" session will rely on Jupyter Notebooks which are targeted for running on Google's Colaboratory Platform or ALCF JupyterHub. The Colab platform gives the user a virtual machine in which to run Python codes including machine learning codes. The VM comes with a preinstalled environment that includes most of what is needed for these tutorials.

The other sessions involve Python scripts executed on the Aurora and AI Testbed platforms at ALCF.

Reservations

  • Queue:
    • Daytime reservations: -q ATPESC
    • Evening reservations: -q ATPESC-Night
    • Outside of reservations: -q debug or -q prod (more info)
  • Project/Allocation: ATPESC2026 (-A ATPESC2026)
  • Shared directories:
    • Aurora: /flare/ATPESC2026
    • Polaris: /eagle/ATPESC2026
  • ALCF Unix Groups: ATPESC2026

Using Google Colab

Google Colab involves running Jupyter notebooks.

Do the following before you come to the tutorial:

  • You need a Google Account to use Colaboratory
  • Go to Google's Colaboratory Platform
  • You should see this page start_page
  • Now you can open the File menu at the top left and select Open Notebook which will open a dialogue box.
  • Select the GitHub tab in the dialogue box.
  • From here you can enter the url for the github repo: https://github.com/argonne-lcf/ATPESC_MachineLearning and hit <enter>. open_github
  • This will show you a list of the Notebooks available in the repo. When you select a notebook from this list it will create a copy for you in your Colaboratory account (all *.ipynb files in the Colaboratory account will be stored in your Google Drive).
  • To use a GPU in the notbook select Runtime -> Change Runtime Type and select an accelerator.

Weights & Biases API key

For the Pre-training on a Supercomputer session, you will need a Wandb api_key. Visit https://docs.wandb.ai/quickstart/ to sign-up and get the key.

Languages

Python

55.0%

Jupyter Notebook

24.8%

HTML

16.1%

Shell

4.1%