Yeah, I fixed it by doing
features.requires_grad = True
outputs = self.net(features)
derivatives = torch.autograd.grad(
    outputs=outputs,
    inputs=features,
    grad_outputs=torch.ones_like(outputs),
    create_graph=True
)[0]

In case anyone runs into the same problem.