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
| File | Mode | Size |
|---|---|---|
| __init__.py | -rw-r--r-- | 575 bytes |
| ad.py | -rw-r--r-- | 24.4 KB |
| batching.py | -rw-r--r-- | 15.4 KB |
| masking.py | -rw-r--r-- | 13.4 KB |
| parallel.py | -rw-r--r-- | 4.8 KB |
| partial_eval.py | -rw-r--r-- | 27.0 KB |
| pxla.py | -rw-r--r-- | 37.2 KB |
| sharded_jit.py | -rw-r--r-- | 9.8 KB |
| xla.py | -rw-r--r-- | 38.6 KB |
Computing file changes ...
