Simulation gym for AI agents