mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-26 18:46:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
17,948 Commits 508 Branches 263 Tags
Commit Graph

2 Commits

Author SHA1 Message Date
pizzud
04def0b6ab lazy_loader_module: Move to new internal_test_util directory. 
Now we no longer need to mess with sys.path in lazy_loader_test.

PiperOrigin-RevId: 515674188
 2023-03-10 10:29:33 -08:00
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