search

pytorch / torchtitan

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

Update pipelining import after change on pytorch #356

Closed wconstab closed 1 month ago

wconstab commented 1 month ago

Stack from ghstack (oldest at bottom):

APIs conform to the pytorch rules. This PR should be able to land safely after tonight's nightly pytorch build which includes the above PR.