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

357 Commits

Author SHA1 Message Date
George Necula
b18a4d8583 Disabled tests known to fail on Mac, and optionally slow tests. 
Issue: #2166

Added JAX_SKIP_SLOW_TESTS environment variable to skip tests known
to be slow.
 2020-02-05 18:02:56 +01:00
George Necula
272620e66c Added note to CHANGELOG.md 2020-02-04 10:22:54 +01:00
Peter Hawkins
91cd20b173
Update documentation and changelog to mention DLPack and array interface support. (#2134) 2020-01-31 11:15:04 -05:00
Roy Frostig
664a4e123d
VJP of cond, via partial eval + transpose (#2091) 
VJP (grad) of lax.cond, via partial eval + transpose


Co-authored-by: Matthew Johnson <mattjj@google.com>
 2020-01-30 15:03:00 -08:00
Peter Hawkins
991324f8df
Increase minimum jaxlib version to 0.1.38. (#2120) 2020-01-29 14:16:58 -05:00
Peter Hawkins
0fe601227f Update README.md and CHANGELOG.md. 2020-01-28 10:00:32 -05:00
Skye Wanderman-Milne
19fb494adb
Add jax changelog (#2022) 2020-01-17 11:34:59 -08:00