Peter Hawkins 68f2cb4491 Implement JVP rule for reduce_prod(). ...

This is sufficient to compute first-order derivatives of a product reduction (although not second-order derivatives because there is no JVP for reduce-window-prod).

2019-05-05 14:37:25 -04:00

..

__init__.py

expose lax._safe_mul again (c.f. #608)

2019-04-13 08:18:57 -07:00

lax_control_flow.py

instantiate consts when tracing while's body_fun

2019-04-30 11:52:43 -07:00

lax_parallel.py

Fix a couple of broken error messages

2019-04-17 23:10:59 -07:00

lax.py

Implement JVP rule for reduce_prod().

2019-05-05 14:37:25 -04:00