Dimitrios Tsaras

Missed to include the versions: 0.4.0 1.26.2 3.10.13 (main, Sep 11 2023, 08:16:02) [Clang 14.0.6 ] darwin

Ok thanks that's helpful! Also, I wonder if we could create dense stacks of the nonuniform shaped tensors by batching them as nested tensors instead?

I wonder if we encounter such data maybe we can automatically pad the data and return a padding mask along the dynamic dimensions. This can be the fallback when the...

I think this feature works pretty well, it would be nice to merge it to the main branch.

I will start testing this today and provide you with some feedback!