This issue tracks the development of quantization for RL using the Megatron training backend and vLLM inference backend in SkyRL. ## FP8 Training - [ ] Supporting Deepseek style blockwise (128x128) full FP8 training on Hopper + Blackwell: https://github.com/NovaSky-AI/SkyRL/pull/1898 - [ ] Enabling FP8 KV cache in vLLM: https://github.com/NovaSky-AI/SkyRL/pull/1899 - [ ] Enabling BF16 LoRA training with FP8 base weights - [ ] Support MXFP8 quantization on experts: https://github.com/NovaSky-AI/SkyRL/pull/1940 - [ ] Use per-token activation scaling in FP32 (as in https://arxiv.org/pdf/2603.24477v2) - [ ] Bump TransformerEngine version from 2.11 -> 2.16.1 - [ ] validate use of dequantized backward in TransformerEngine per (https://humansand.ai/blog/nvfp4-rl?v=3, https://github.com/NVIDIA/TransformerEngine/pull/2644) - use `NVTE_BACKWARD_OVERRIDE=dequantized` - [ ] Improve support for specifying which layers should be quantized ## INT4 QAT Training - [x] Support LoRA only INT4 Fake QAT training (W4A16 serving, fake INT4 quant training): https://github.com/NovaSky-AI/SkyRL/pull/1862 ## NVFP4 Training - [ ] Support NVFP4 Training with 4 over 6 weights/activations per https://humansand.ai/blog/nvfp4-rl?v=3
This issue tracks the development of quantization for RL using the Megatron training backend and vLLM inference backend in SkyRL.
FP8 Training
NVTE_BACKWARD_OVERRIDE=high_precision|dequantizedNVIDIA/TransformerEngine#2644) - useNVTE_BACKWARD_OVERRIDE=dequantizedINT4 QAT Training
NVFP4 Training