It’s the first time that I see this one.
Are you using the jit?
Otherwise, I’m not sure to understand the point of having a parameters that you do not use? You can do return zeros * self._par such that your output at least requires gradients.