AI for complex scientific simulations