Optimization Of Cosmological Simulations With Artificial Intelligence