Solved by soulitzer in post #2 
              
              
                Yep. If you want to check if something is differentiable you can just call it with a tensor that requires grad and it’ll spit out a warning if the function does not support autograd