Training a forward dynamics model of a physical system using neural network

Question

I am trying to predict the forward dynamics of a physical system using neural networks. The equation is f(X_t,A_t) = (X_{t+1} - X_t)/dt where X is