Activation Checkpointing + FlashAttention + GRPO