Developing machine learning techniques for non-equilibrium dynamics