-
|
Hello JAX team, I have read through the tutorial part of the JAX document and am still have some questions about how AD is implemented in JAX. Does JAX support Newton-Raphson's reverse AD? If it is not currently supported, will it be developed in the future? @jakevdp |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 1 reply
-
|
Yes. How to implement memory-efficient reverse AD for Newton-Raphson is explained in this tutorial/talk. This library provides a convenient API using the same techniques. |
Beta Was this translation helpful? Give feedback.
-
|
Thank you very much for your quick response! It is very helpful to me! |
Beta Was this translation helpful? Give feedback.
Yes. How to implement memory-efficient reverse AD for Newton-Raphson is explained in this tutorial/talk. This library provides a convenient API using the same techniques.