Astrophysics Source Code Library

Making codes discoverable since 1999

ASCL Code Record

[ascl:2111.002] JAX: Autograd and XLA

JAX brings Autograd and XLA together for high-performance machine learning research. It can automatically differentiate native Python and NumPy functions. The code can differentiate through loops, branches, recursion, and closures, and it can take derivatives of derivatives of derivatives. JAX supports reverse-mode differentiation (a.k.a. backpropagation) via grad as well as forward-mode differentiation, and the two can be composed arbitrarily to any order.

Code site:
Used in:
Preferred citation method:

Please see citation information here:

Views: 1304

Add this shield to your page
Copy the above HTML to add this shield to your code's website.