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

2 Commits

Author SHA1 Message Date
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