James Bradbury 55d74d8624 add VJP for lax._select_and_gather_add (3rd-order grad of maxpool)

2019-04-20 17:06:56 -07:00

..

__init__.py

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

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

lax_control_flow.py

make a lax package, revert control flow names (#607)

2019-04-12 16:28:40 -07:00

lax_parallel.py

Fix a couple of broken error messages

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

lax.py

add VJP for lax._select_and_gather_add (3rd-order grad of maxpool)

2019-04-20 17:06:56 -07:00