Naive attempt at implementing TTT paper by letting autograd do the heavy lifting - View it on GitHub