AI for data-driven simulations in Physics