Erich Elsen b46bd2301c add support bool identity values

2020-06-29 19:13:41 +01:00

..

__init__.py

add cummax/min to __init__.py

2020-06-28 19:07:27 +01: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

Avoid tuple psum in pmean (#3479)

2020-06-17 21:42:17 -07:00

lax.py

add support bool identity values

2020-06-29 19:13:41 +01:00