pytorch

e9954952 - [spmd] move the param_buffers to the front (#98437)

[spmd] move the param_buffers to the front (#98437) Makes it a bit easy to track the parameter count and corresponding named states Pull Request resolved: https://github.com/pytorch/pytorch/pull/98437 Approved by: https://github.com/mrshenli