Add option for sequence padding with packed sequences

[ahmadki]

We need fixed shapes in order to support torch.compile in dtensor

[azahed98]

@ahmadki Just came across this - I'm currently working on improving dynamic shape support with DTensor in PyTorch. Do you have existing issues filed for the dynamic shape related errors you came across? And are these issues with dynamic shapes + DTensor blocking enabling compile?