~/talk

How Pytorch Powers Training Inference

Talkat AI Infra @Scale 2024 · on Wednesday, 2024-07-31· [large-language-models]
links:none on file
linkedin:kimishpatel
links:none on file
slides:none on file

Shows how PyTorch stacks combine concrete techniques and numbers: torch Titan composes 3D parallelism, activation recomputation and TorchCompile to run model pretraining; torch tune offers single-file, hackable recipes for adapter, quantization-aware and CPU‑offload workflows so a Llama‑27B needs ~56 GB while Llama‑370B can be adapter‑fine‑tuned on two 4090s; distributed tooling scales Mixture‑of‑Experts to 3,000 GPUs and collaborative work yields 10x faster checkpointing for long runs, plus low‑bit quantization and ExecuteTorch for on‑device inference.

AI-written summary · about

  1. 0:00PyTorch and LLMs
  2. 1:09Large scale training
  3. 3:54Distributed training primitives
  4. 6:24TorchTitan pre-training
  5. 8:11Training large models challenges
  6. 8:32Fine-tuning LLMs
  7. 13:10TorchTune fine-tuning
  8. 15:59Fine-tuning and multimodality
  9. 16:41On-device inference for LLMs
  10. 19:30Quantization and sparsity
  11. 20:57PyTorch inference tools
  12. 22:44Generative AI future
~/search
type to search 6,249 conferences, 138,987 speakers and 185,172 talks
~/help
cmdctrl k
search every conference, speaker and series
/
focus the filter search
h
home
c
conferences
s
speakers
e
series
t
light / dark — the status bar button does the same
?
this help
esc
close this, then clear every filter — or just the search, while you are typing in it

Ignored while you are typing. Pages without a filter rail have no search to focus.