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 |
|---|---|---|
| experimental | ||
| interpreters | ||
| lax | ||
| lib | ||
| nn | ||
| numpy | ||
| ops | ||
| scipy | ||
| third_party | ||
| tools | ||
| BUILD | -rw-r--r-- | 2.0 KB |
| __init__.py | -rw-r--r-- | 813 bytes |
| abstract_arrays.py | -rw-r--r-- | 2.4 KB |
| ad_util.py | -rw-r--r-- | 1.7 KB |
| api.py | -rw-r--r-- | 70.0 KB |
| api_util.py | -rw-r--r-- | 3.3 KB |
| config.py | -rw-r--r-- | 4.0 KB |
| core.py | -rw-r--r-- | 32.5 KB |
| custom_derivatives.py | -rw-r--r-- | 22.9 KB |
| dlpack.py | -rw-r--r-- | 1.9 KB |
| dtypes.py | -rw-r--r-- | 7.3 KB |
| flatten_util.py | -rw-r--r-- | 1.2 KB |
| lax_linalg.py | -rw-r--r-- | 32.2 KB |
| lax_reference.py | -rw-r--r-- | 14.2 KB |
| lazy.py | -rw-r--r-- | 8.8 KB |
| linear_util.py | -rw-r--r-- | 8.1 KB |
| pprint_util.py | -rw-r--r-- | 1.9 KB |
| profiler.py | -rw-r--r-- | 2.7 KB |
| random.py | -rw-r--r-- | 42.1 KB |
| test_util.py | -rw-r--r-- | 24.3 KB |
| tree_util.py | -rw-r--r-- | 9.6 KB |
| util.py | -rw-r--r-- | 5.3 KB |
| version.py | -rw-r--r-- | 599 bytes |
Computing file changes ...
