saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/torch/distributed/tensor/parallel
History
Aaron Orenstein c95efc37ba PEP585 update - torch/distributed/tensor (#145141) 
See #145101 for details.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145141
Approved by: https://github.com/bobrenjc93
2025-01-18 20:01:59 +00:00
..
__init__.py
_data_parallel_utils.py
_utils.py
api.py PEP585 update - torch/distributed/tensor (#145141) 2025-01-18 20:01:59 +00:00
ddp.py PEP585 update - torch/distributed/tensor (#145141) 2025-01-18 20:01:59 +00:00
fsdp.py PEP585 update - torch/distributed/tensor (#145141) 2025-01-18 20:01:59 +00:00
input_reshard.py
loss.py PEP585 update - torch/distributed/tensor (#145141) 2025-01-18 20:01:59 +00:00
style.py PEP585 update - torch/distributed/tensor (#145141) 2025-01-18 20:01:59 +00:00