James Bradbury f78ccf1c39

Fix tuple() in reduce_window padding (#3748) ...

* Fix tuple() in reduce_window padding

* Update lax.py

2020-07-13 18:16:11 -07:00

..

__init__.py

Reimplement argmin/argmax using a single pass variadic reduction. (#3611)

2020-07-01 11:01:22 -04:00

lax_control_flow.py

introduce custom typecheck rules, implement them for cond and scan

2020-06-26 15:31:04 -07:00

lax_fft.py

deflake jax/lax & add to flake8 check (#3310)

2020-06-04 13:50:44 -07:00

lax_parallel.py

fix ppermute test bugs found by @jekbradbury (#3675)

2020-07-07 00:30:08 -07:00

lax.py

Fix tuple() in reduce_window padding (#3748)

2020-07-13 18:16:11 -07:00