attractor-learning attractor-network computational-neuroscience diffusion head-direction iclr iclr2024 langevin-dynamics langevin-mc learning path-integration pytorch reactivation recurrent-networks recurrent-neural-nets recurrent-neural-network recurrent-neural-networks replay rnn spatial-navigation