Home

TorchRelay logo

This is a lightweight packages that:

  • implements backward-compatible functions that mimic the modern PyTorch API (linalg, fft, movedim, ...)
  • implements extended variants of functions from the PyTorch functional API (pad, one_hot, ...)
  • implements a range of helpers compatible with TorchScript.

This package is under active development and still largely untested.