spmd_types: A type system for distributed (SPMD) tensor computations in PyTorchgithub.com/meta-pytorch2 pointsmatt_d2 months ago