mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
7,014 Commits 504 Branches 263 Tags
Commit Graph

2 Commits

Author SHA1 Message Date
Roy Frostig
4c179d1b66 register the lax module for stack trace filtering 2021-02-22 21:11:41 -08:00
Peter Hawkins
10b7d7d7c2 Move implementation of jax.lax into jax._src.lax. 
Remove lax_ prefixes from jax/_src/lax filenames, since they aren't needed any longer to avoid name conflicts.
 2020-10-17 16:09:21 -04:00