jax

18623265 - [doc] update docs for eig() autodiff