Equivariant Neural Simulators for Stochastic Spatiotemporal Dynamics

Neural Information Processing Systems 

However, for many applications this formulation is insufficient.