Skip to content

Releases: piyueh/TorchSWE

torchswe-v0.1

03 Sep 19:03
8984aa9
Compare
Choose a tag to compare
  • Single-node version is working with all four backends: legate, cupy, pytorch, and numpy
  • MPI version is working with MPI+CuPy and MPI+PyTorch