Loreon
Labs
Platform
Docs
Home
Ecosystems
Other
torchtitan
A native PyTorch Library for large model training
Other
Emerging
GitHub
Stars
—
Forks
—
Contributors
8
Last push
24mo ago
Recent commits
Latest commits.
Added type annotations and more stylistic changes
040ea1d
Andrew Gu
24mo ago
Renamed parallel styles for transformer block weights
0929280
Andrew Gu
24mo ago
compiled RMSNorm
7afe902
Tianyu Liu
24mo ago
Refactored activation checkpointing
bc3ec02
Andrew Gu
24mo ago
Made some stylistic changes to `apply_dp`
c7a6a3e
Andrew Gu
24mo ago
Reordered TP parallel plan to follow execution order
f0ca3e8
Andrew Gu
24mo ago
Removed `_experimental_support_context_fn_in_torch_utils_checkpoint`
2f5285b
Andrew Gu
24mo ago
switch float8 logic from Float8DynamicLinear to Float8Linear (#436)
3fca883
Vasiliy Kuznetsov
24mo ago
Top contributors
Builders behind this project.
wanchaol
48 commits
tianyu-l
47 commits
wconstab
41 commits
lessw2020
31 commits
gnadathur
14 commits
wz337
14 commits
drisspg
5 commits
kwen2501
5 commits