Peter Hawkins 9ab88071a7 Avoid loading scipy eagerly. ...

scipy accounts for around 400ms of the 900ms of JAX's import time. By
loading scipy lazily, we can improve the timing of `import jax` down to
about 500ms.

2022-10-12 19:51:09 +00:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

functions.py

Avoid loading scipy eagerly.

2022-10-12 19:51:09 +00:00

initializers.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00