Shampoo: Preconditioned Stochastic Tensor Optimization