[Deprecation] update deprecated pytorch `custom_fwd` and `custom_bwd` function from `torch.cuda.amp`#560

Jonathanjordan21 wants to merge 4 commits intostate-spaces:mainstate-spaces/mamba:mainfrom

Jonathanjordan21:deprecation/torch_cuda_ampJonathanjordan21/mamba:deprecation/torch_cuda_ampCopy head branch name to clipboard