Correctly get parameters with unsharded grads and parameters to register sharded and sync'ed grads #1643
Azure Pipelines / lightning-thunder (ipynb) (jupyter ubuntu22.04 | cuda 12.1 | torch 2.4.0)
succeeded
Jan 14, 2025 in 8m 8s
jupyter ubuntu22.04 | cuda 12.1 | torch 2.4.0 succeeded
Loading