spmd_types. This module defines a type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosystem.

github.com/meta-pytorch/spmd_types

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.