This seems to be due to a bug in how we are interfacing with autograd, or -- less likely -- a bug in…

This seems to be due to a bug in how we are interfacing with autograd, or -- less likely -- a bug in…: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). If the user provides a quantum function func with multiple input arguments in its signature, we can only…

The task

If the user provides a quantum function `func` with multiple input arguments in its signature, we can only successfully call `qml.jacobian(func, argnum=0)`, `qml.jacobian(func, argnum=1)`, etc., but not `qml.jacobian(func, argnum=[0,1])` (raises some error inside autograd).

Part of FineEnvs/MiMo-V2.6-RL-harbor-code.