Peter Hawkins d658108d36 Fix type errors with current mypy and NumPy. ...

Enable type stubs for jaxlib.

Fix a nondeterminism problem in jax2tf tests.

2021-06-24 10:51:06 -04:00

..

__init__.py

Move most of the implementation of jax.scipy into jax._src.scipy.

2020-10-16 17:04:25 -04:00

_lbfgs.py

Fix type errors with current mypy and NumPy.

2021-06-24 10:51:06 -04:00

bfgs.py

Update Jax linesearch to behave more like Scipy

2021-05-26 12:49:56 +01:00

line_search.py

Merge branch 'master' into master

2021-06-02 01:05:47 -07:00

minimize.py

add L-BFGS optimizer

2021-05-19 19:46:11 +02:00