Switch to torch.compile for dropout for torch v>2.2

Switch to torch.compile for dropout for torch v>2.2 #2916