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-oss: Agentic RL Environments (MiMo RL release). If the user provides a quantum function func with multiple input arguments in its signature, we can only successfully call…

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). Since we've…

Part of XiaomiMiMo/MiMo-V2.6-RL-oss.