Revision 1cf708ea77fae0fc1418e0b944f0115646f7f2ec authored by Stephan Hoyer on 04 April 2020, 22:55:46 UTC, committed by GitHub on 04 April 2020, 22:55:46 UTC
* Support pytrees in jax.scipy.linalg.cg

Ideally there would be an easier way to write this, but for now this will do.

* Fixup test
1 parent 2c4ced2
History
File Mode Size
__init__.py -rw-r--r-- 673 bytes
fft.py -rw-r--r-- 7.3 KB
lax_numpy.py -rw-r--r-- 122.1 KB
linalg.py -rw-r--r-- 11.6 KB
vectorize.py -rw-r--r-- 8.8 KB

back to top