Revision 824ac86620572285b86dd09529f8869ef36883ad authored by Peter Hawkins on 03 April 2020, 19:39:56 UTC, committed by GitHub on 03 April 2020, 19:39:56 UTC
* Reimplement np.cumsum and np.cumprod in terms of a parallel prefix scan. Unlike the existing implementation based on lax.reduce_window, this implementation is O(n log n) instead of O(n^2) and is arbitrarily differentiable. Fixes #1212, #2418, #2542. May help with issue #2380. * Relax gradient test tolerance.
1 parent 192e908
| File | Mode | Size |
|---|---|---|
| benchmarks | ||
| build | ||
| cloud_tpu_colabs | ||
| design_notes | ||
| docs | ||
| examples | ||
| images | ||
| jax | ||
| jaxlib | ||
| tests | ||
| .gitignore | -rw-r--r-- | 183 bytes |
| .readthedocs.yml | -rw-r--r-- | 504 bytes |
| .travis.yml | -rw-r--r-- | 1.9 KB |
| CHANGELOG.md | -rw-r--r-- | 92 bytes |
| CONTRIBUTING.md | -rw-r--r-- | 581 bytes |
| LICENSE | -rw-r--r-- | 11.1 KB |
| LICENSE_SHORT | -rw-r--r-- | 575 bytes |
| README.md | -rw-r--r-- | 19.5 KB |
| WORKSPACE | -rw-r--r-- | 2.4 KB |
| mypy.ini | -rw-r--r-- | 292 bytes |
| pylintrc | -rw-r--r-- | 1.6 KB |
| setup.py | -rw-r--r-- | 1.1 KB |
Computing file changes ...
