-
|
Hi there, I want to apply JIT compilation to some numpy code which has nothing to do with auto differentiation. Do I need to tell Jax I don't need the gradient information to eliminate some tracking overhead? |
Beta Was this translation helpful? Give feedback.
Answered by
mattjj
May 20, 2022
Replies: 1 comment
-
|
No need to do anything. There's no autodiff overhead unless you actually use autodiff. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
nalzok
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
No need to do anything. There's no autodiff overhead unless you actually use autodiff.