MyNixOS website logo
Description

Type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosystem.

Metadata

Version

0.2.5

Maintainers (1)

Platforms (31)

    Darwin
    FreeBSD
    Linux
    Windows
Show all
  • aarch64-darwin
  • aarch64-freebsd
  • aarch64-linux
  • aarch64-windows
  • arc-linux
  • armv5tel-linux
  • armv6l-linux
  • armv7a-linux
  • armv7l-linux
  • i686-freebsd
  • i686-linux
  • i686-windows
  • loongarch64-linux
  • m68k-linux
  • microblaze-linux
  • microblazeel-linux
  • mips-linux
  • mips64-linux
  • mips64el-linux
  • mipsel-linux
  • powerpc-linux
  • powerpc64-linux
  • powerpc64le-linux
  • riscv32-linux
  • riscv64-linux
  • s390-linux
  • s390x-linux
  • sh4-linux
  • x86_64-freebsd
  • x86_64-linux
  • x86_64-windows