Category "automatic-differentiation"

Jacobian diagonal computation in JAX

I was wondering how we could use jax (https://github.com/google/jax) to compute a mapping of the derivative. That is to say : we have a vector and we want t

Automatic differentiation using expression templates c++

Introduction I am trying to learn about expression templates because it seems to be a very powerful technique for a wide range of calculations. I looked at diff

Differentiation of an improper integral using JAX and SciPy

I provide a simple code example of a failed attempt to use JAX to automatically differentiate through an improper integral function making use of SciPy's quad()