Everyone's favorite one-notebook minimal pytorch implementation of TensorNetwork for Machine Learning paper by Google.

Following the paper, we implement Matrix Product State (MPS) as a pytorch module and test it on MNIST.

If you you liked this implementation you might check my paper inspired by this one.