Custom collate_fn for Variable-Length Sequences | AlgoML Prep