**Describe the bug** Test failing: llm_grpo_llama3_2_1b_instruct_1n8g_megatron_generation Likely due to the mcore bump 8ef0de9dd1647753dbc89aff822d3c06e196020e The test is now timing out. **Expected behavior** Test should pass ideally without timing out.