A Python/PyTorch implementation of a 2D elliptic multigrid solver for Poisson or Helmholtz like equations - View it on GitHub