search

pytorch / torchtitan

A native PyTorch Library for large model training
BSD 3-Clause "New" or "Revised" License
2.29k stars 170 forks source link

try nvidia-cuda #331

Closed wconstab closed 4 months ago

wconstab commented 4 months ago

Stack from ghstack (oldest at bottom):