Using Machine Learning to Accelerate PDE solvers