Description
Type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosystem.
Type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosystem.
0.2.5aarch64-darwinaarch64-freebsdaarch64-linuxaarch64-windowsarc-linuxarmv5tel-linuxarmv6l-linuxarmv7a-linuxarmv7l-linuxi686-freebsdi686-linuxi686-windowsloongarch64-linuxm68k-linuxmicroblaze-linuxmicroblazeel-linuxmips-linuxmips64-linuxmips64el-linuxmipsel-linuxpowerpc-linuxpowerpc64-linuxpowerpc64le-linuxriscv32-linuxriscv64-linuxs390-linuxs390x-linuxsh4-linuxx86_64-freebsdx86_64-linuxx86_64-windows