Roy Frostig 4c179d1b66 register the lax module for stack trace filtering

2021-02-22 21:11:41 -08:00

..

__init__.py

register the lax module for stack trace filtering

2021-02-22 21:11:41 -08:00

control_flow.py

Merge pull request #5659 from google:cond-doc-wrap

2021-02-05 21:40:52 -08:00

fft.py

[avals with names] Revise aval constructor call sites to use a new aval.update method

2021-01-27 15:14:02 -08:00

lax.py

Implement lax.pad batching rule for batched padding values.

2021-02-17 13:30:20 -05:00

linalg.py

Improve error reporting for complex eigh on TPU

2021-02-05 10:11:10 -08:00

other.py

[JAX] Move jax.util to jax._src_util.

2021-01-11 14:21:07 -08:00

parallel.py

Rollback of #5702 due to internal breakage.

2021-02-17 07:32:09 -08:00