dtensor

Here are 3 public repositories matching this topic...

A native PyTorch backend for AWS Trainium with support for eager execution, torch.compile, distributed training, and custom NKI kernels

training aws machine-learning deep-learning inference pytorch nki fsdp aws-neuron trainium torch-compile dtensor

Fork of PyTorch TorchTitan that adds AWS Trainium support with TorchNeuron

training aws machine-learning deep-learning pytorch nki fsdp aws-neuron trainium torch-compile torchtitan dtensor

Syzkaller-style fuzzers for PyTorch distributed: c10d/Gloo collectives, DTensor, and distributed checkpointing

distributed-systems pytorch fuzzing differential-testing gloo syzkaller dtensor

To associate your repository with the dtensor topic, visit your repo's landing page and select "manage topics."