Evenly distribute complete chunks/batches across partial dataset splits#3476

laggui merged 1 commit intomaintracel-ai/burn:mainfrom

feat/data/split-chunkstracel-ai/burn:feat/data/split-chunksCopy head branch name to clipboard