Capturing dynamical correlations using implicit neural representations