Peter Hawkins e4a7d30741 Fix batching rule.

2019-08-15 11:42:08 -04:00

..

__init__.py

Check for failure of LAPACK calls and return arrays full of NaNs if they occur.

2019-05-29 16:50:27 -04:00

lax_control_flow.py

More concurrency fixes.

2019-08-09 15:08:26 -04:00

lax_fft.py

Adding vmap/batching support to FFT.

2019-07-19 14:15:33 +01:00

lax_parallel.py

Fix documentation for jax.lax.ppermute

2019-08-08 16:24:46 -07:00

lax.py

Fix batching rule.

2019-08-15 11:42:08 -04:00