Support dynamic dimensions

[BinaryWarlock]

Is your feature request related to a problem? Please describe. I'd like to use dynamic batch sizes to actually batch properly.

Describe the solution you'd like Dynamic dimension support

Describe alternatives you've considered Baking in the batch size with nnx prepare isn't remotely ideal, since if my dataset is not a multiple of the batch size I'd have to pad it somehow and that's pretty awful.