You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
When using custom_jvp or custom_vjp, don't use nondiff_argnums for array-valued arguments. It'll often lead to "encountered an unexpected tracer" errors.
But we should raise a better error, and make the docs more discoverable (and clearer).
When using
custom_jvp
orcustom_vjp
, don't usenondiff_argnums
for array-valued arguments. It'll often lead to "encountered an unexpected tracer" errors.But we should raise a better error, and make the docs more discoverable (and clearer).
Here's a repro from a user:
The text was updated successfully, but these errors were encountered: