a lightweight simulation environment for understanding AI generalization.