-
Notifications
You must be signed in to change notification settings - Fork 0
-
Notifications
You must be signed in to change notification settings - Fork 0
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
```python #1
Comments
Open
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Notice especially this part that points to the user's function
body
:But if we do the analogous thing with
fori_loop
:Oops, we point to an internal function name rather than the user's:
We should probably improve that to point to the user function name, like we do with
scan
, rather than our internal wrapper.Originally posted by @mattjj in google/jax#23637
The text was updated successfully, but these errors were encountered: