Surrogate Models for RBC TODO test all scripts ask codex for code quality and suggestions notebooks experiment configs sweeps ruff check slurm scripts *.log file check sweep params Update packages