nixsearch
nixsearch / pkg / python38Packages.torchgpipe / 0.0.7

python38Packages.torchgpipe 0.0.7

GPipe implemented in Pytorch and optimized for CUDA rather than TPU
Linux and macOS last changed

Pin this version

$ devbox add python38Packages.torchgpipe@0.0.7
systemsaarch64-darwin, x86_64-darwin, x86_64-linux commita801c31fa47618d4fd27ee4c15d4c58b5ae60cae 
$ nix shell github:NixOS/nixpkgs/a801c31fa47618d4fd27ee4c15d4c58b5ae60cae#python38Packages.torchgpipe

Per system

SystemAttributeCommitDateOutputsFlags
aarch64-darwin
macOS · arm64		 python38Packages.torchgpipe a801c31fa476
x86_64-darwin
macOS · x86-64		 python38Packages.torchgpipe a801c31fa476
x86_64-linux
Linux · x86-64		 python38Packages.torchgpipe a801c31fa476
all 2 releases older: 0.0.5 →

JSON: /v2/pkg?name=python38Packages.torchgpipe · /v2/resolve?name=python38Packages.torchgpipe&version=0.0.7