Pytorch implementation of preconditioned stochastic gradient descent (af...
Tensorflow implementation of preconditioned stochastic gradient descent