feat: Add attention_backend config support for Megatron policy#1628

Open

sahgerlad wants to merge 7 commits intoNVIDIA-NeMo:mainfrom

sahgerlad:feat/attention-backend-config

Commits on Feb 4, 2026

feat: Add attention_backend config support for Megatron policy
sahgerlad
committed
test: Add unit test for attention_backend config in Megatron policy
sahgerlad
committed

Commits on Mar 21, 2026

Merge branch 'main' into feat/attention-backend-config
sahgerlad
authored

Commits on Mar 23, 2026

Update nemo_rl/models/megatron/setup.py

sahgerlad
and
yuki-97
authored
Merge branch 'main' into feat/attention-backend-config
sahgerlad
authored
Fix unit test to use bf16 since flash attention doesn't support FP32; remove env var that causes issues
sahgerlad
committed

Commits on Mar 24, 2026

Merge branch 'main' into feat/attention-backend-config
sahgerlad
authored