Machine Learning Out-of-Equilibrium Phases of Matter