We plan to increase the number of symbolic derivatives that we support, but we don’t have a specific timeline for when we’ll reach full coverage. In the meantime, autograd still works for JIT modules, so as long as you set requires_grad on your input tensors you will be able call backward() like in Python.