No, not using JIT.
I need a parameter because otherwise PyTorch complains:
ValueError: optimizer got an empty parameter list

Output already requires gradients, since I initialise it as such.
Multiplying zeros by the parameter didn’t help, same error.