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

3 Commits

Author SHA1 Message Date
Jake VanderPlas
f090074d86 Avoid 'from jax import config' imports 
In some environments this appears to import the config module rather than
the config object.
 2024-04-11 13:23:27 -07:00
Jake VanderPlas
fbe4f10403 Change to simpler import for jax.config 2023-04-21 11:51:22 -07:00
Peter Hawkins
a3b262c379 Use the traceback of the call site when assigning a source location to an inlined function. 
Improves but does not completely fix https://github.com/google/jax/issues/15663 . The non-inlined case still has similar problems.
 2023-04-19 13:56:53 -04:00