Peter Hawkins 3ddcec27f2 Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00

..

__init__.py

register the lax module for stack trace filtering

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

control_flow.py

Cleanup: use pep 448 unpacking to simplify some code

2021-07-12 16:30:53 -07:00

fft.py

Move jax.api to jax._src.api.

2021-04-13 09:43:24 -07:00

lax.py

Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00

linalg.py

Test for gtsv2 attr on cusparse.

2021-06-23 14:22:08 -07:00

other.py

Add missing typing.Optional type annotations to function parameters.

2021-05-27 20:10:23 -07:00

parallel.py

Add axis and tiled options to lax.all_gather.

2021-07-15 04:22:36 -07:00