pytorch-lightning

4a548c96 - fix: restore cached AMP step context after no_grad workaround (#21616)

fix: restore cached AMP step context after no_grad workaround (#21616) * fix: restore cached AMP step context after no_grad workaround * chore: trigger ci * chore: trigger ci * test: add CUDA coverage for AMP no_grad cache handling