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
After #897, vjp does not accept int input. Because ravel_pytree uses vjp for ravel/unravel logic, something like ravel_pytree([1]) does not work anymore. I would be nice if we are able to disable this check in vjp. Probably we can provide an optional kwargs disable_check_inexact_input in vjp function to maintain this behaviour.
After #897,
vjp
does not accept int input. Becauseravel_pytree
usesvjp
for ravel/unravel logic, something likeravel_pytree([1])
does not work anymore. I would be nice if we are able to disable this check invjp
. Probably we can provide an optional kwargsdisable_check_inexact_input
invjp
function to maintain this behaviour.cc @hawkinsp @neerajprad
The text was updated successfully, but these errors were encountered: