autodiff support for jax.numpy.linalg.eig

jax-ml / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

http://jax.readthedocs.io/

Apache License 2.0

30k stars 2.75k forks source link

Open froystig opened 4 years ago

froystig commented 4 years ago

Note that eigh is already taken care of.

tomsturges commented 8 months ago

There seems to be an implementation here