pytorch/torch/distributed/tensor/examples
2025-02-28 07:35:56 +00:00
..
comm_mode_features_example.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
convnext_example.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
flex_attention_cp.py [2/N][cp][example] flex attention in context parallel (backward pass) (#146397) 2025-02-06 19:50:02 +00:00
torchrec_sharding_example.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
visualize_sharding_example.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00