Pytorch implementation of Ganguli et. al paper "A unified theory for the origin of grid cells through the lens of pattern formation"