Commits · main · Simulation and Data Lab Applied Machine Learning / PyTorch at JSC · GitLab

Aug 16, 2025
- Update cache directories · a68bd66a
  Jan Ebert authored 1 month ago
  a68bd66a
Aug 14, 2025
- Be more specific about CPU cores · 5dd97661
  Jan Ebert authored 1 month ago
  5dd97661
Jul 28, 2025
- Improve IPv6 warning explanation · 072873fc
  Jan Ebert authored 1 month ago
  
  072873fc
Jun 04, 2025
- Revert "Use meta device initialization" · b5265637
  Jan Ebert authored 3 months ago
  b5265637
- Fix wrap policy usage · 2090b798
  Jan Ebert authored 3 months ago
  2090b798
- Fix function call · 7da56b3e
  Jan Ebert authored 3 months ago
  
  7da56b3e
Jun 03, 2025
- Use meta device initialization · 528e690f
  Jan Ebert authored 3 months ago
  
  528e690f
Jun 02, 2025
- Use FSDP2 · 7e183b8c
  Jan Ebert authored 3 months ago
  7e183b8c
- Allow using Vision Transformer · e90d4094
  Jan Ebert authored 3 months ago
  
  e90d4094
- Refactor model distribution · 48ba71ec
  Jan Ebert authored 3 months ago
  
  48ba71ec
- Allow specifying image size · 05fba361
  Jan Ebert authored 3 months ago
  
  05fba361
- Fix required PyTorch version · 4e7a8f6b
  Jan Ebert authored 3 months ago
  
  4e7a8f6b
Mar 06, 2025
- Add HSDP description · f3e90b24
  Jan Ebert authored 6 months ago
  
  f3e90b24
- Add optional HSDP support · 99836847
  Jan Ebert authored 6 months ago
  99836847
- Clean up whitespace · 3fe25198
  Jan Ebert authored 6 months ago
  
  3fe25198
- Explain number of CPUs per task · f3b4d087
  Jan Ebert authored 6 months ago
  
  f3b4d087
Feb 20, 2025
- Fix deprecation warnings · 6303cd82
  Jose Ignacio Robledo authored 7 months ago and Jan Ebert committed 7 months ago
  6303cd82
Feb 17, 2025
- Update required `torchrun_jsc` version · c40183f1
  Jan Ebert authored 7 months ago
  
  c40183f1
Feb 14, 2025
- Do not promote module usage for fixing `torchrun` · 18955298
  Jan Ebert authored 7 months ago
  18955298
Feb 11, 2025
- Mention command to check compute capability · 15a5a05c
  Jan Ebert authored 7 months ago
  15a5a05c
Nov 14, 2024
- Add code for later PyTorch versions · ceb61d4f
  Jan Ebert authored 10 months ago
  ceb61d4f
- Include singular loading function · b4c87a7f
  Jan Ebert authored 10 months ago
  b4c87a7f
- Link to `torchrun_jsc` repo · c0bff5b5
  Jan Ebert authored 10 months ago
  
  c0bff5b5
Nov 03, 2024
- Improve activation script resiliency · 3ca91f8a
  Jan Ebert authored 10 months ago
  3ca91f8a
- Fix train step logging · 871386bd
  Jan Ebert authored 10 months ago
  
  871386bd
- Do not shuffle test/validation splits · 20148fbf
  Jan Ebert authored 10 months ago
  20148fbf
Oct 25, 2024
- Rename distributed checkpointing import · 68e35585
  Jan Ebert authored 10 months ago
  68e35585
Oct 24, 2024
- Mention `TRITON_HOME` · 06888277
  Jan Ebert authored 10 months ago
  
  06888277
Oct 23, 2024
- Add PyTorch kernel cache path · ddde4efe
  Jan Ebert authored 11 months ago
  
  ddde4efe
Oct 21, 2024
- Improve checking whether command exists · 110c0a66
  Jan Ebert authored 11 months ago
  
  110c0a66
Oct 16, 2024
- Specify Gloo backend for CPU · bd050edf
  Jan Ebert authored 11 months ago
  bd050edf
- Properly shut down distributed process group · 1721a1aa
  Jan Ebert authored 11 months ago
  1721a1aa
- Fix typo · ac788119
  Jan Ebert authored 11 months ago
  
  ac788119
Oct 10, 2024
- Fix Gloo name · fa99171e
  Jan Ebert authored 11 months ago
  
  fa99171e
Oct 09, 2024
- Also scale other optimizer parameters · c941cc9c
  Jan Ebert authored 11 months ago
  c941cc9c
- Fix awkward script exits · 373c9406
  Jan Ebert authored 11 months ago
  373c9406
Sep 18, 2024
- Add section on Apptainers · fbe9d453
  Jan Ebert authored Sep 18, 2024
  
  fbe9d453
Aug 16, 2024
- Use newly recommended SMT prevention · f62bb14d
  Jan Ebert authored Aug 16, 2024
  f62bb14d
Aug 14, 2024
- Improve hostname problem description · cad4fbff
  Jan Ebert authored Aug 14, 2024
  cad4fbff
Jul 31, 2024
- Bump `torchrun_jsc` version · e597c354
  Jan Ebert authored Jul 31, 2024
  e597c354