Train batch generic#724

Merged

felipemello1 merged 8 commits into

meta-pytorch:mainfrom

HosseinKaviani-H:TrainBatch_Generic

Jan 26, 2026

Commits on Jan 23, 2026

Add TrainBatch dataclass for universal training batches
Hossein Kavianihamedani
committed
More consice examples
Hossein Kavianihamedani
committed
Move TrainBatch to types.py and update collate imports
Hossein Kavianihamedani
committed
Update TitanTrainer and GRPO main to use TrainBatch
Hossein Kavianihamedani
committed

Commits on Jan 24, 2026

Update test scripts to use TrainBatch
Hossein Kavianihamedani
committed
Update test scripts to use TrainBatch
Hossein Kavianihamedani
committed

Commits on Jan 26, 2026