Enable training on variable-sized batches#42

Open

hanaol wants to merge 2 commits intomainfrom

hanaol/batch-variable-size

Commits on Dec 11, 2025

Add support for training on batches with variable data sizes
Hananeh Oliaei
committed
Support heterogeneous batches in loss function and model forward pass
Hananeh Oliaei
committed