Repository navigation

SFT: device-agnostic chunked linear+CE loss and --gradient-checkpointing forwarding - #330

Merged

michaellchung merged 2 commits into

mainfrom

Nov 3, 2026

Merged

SFT: device-agnostic chunked linear+CE loss and --gradient-checkpointing forwarding#330
michaellchung merged 2 commits into
mainfrom
sft-chunked-ce

Commits on Nov 3, 2026
- Add device-agnostic chunked linear+CE SFT loss backend
  
  authored and
  MichaelChung
  committed
  6df2477
- Forward --gradient-checkpointing from the launcher to torch learners
  
  authored and
  MichaelChung
  committed
  128e1ef