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

6 Commits

Author SHA1 Message Date
Peter Hawkins
db2e91eba2 Move jax.test_util to jax._src.test_util. 
Add forwarding shims for names used by external clients of JAX in practice.

PiperOrigin-RevId: 398721725
 2021-09-24 07:02:49 -07:00
Peter Hawkins
9f083d11da Use jax.* APIs rather than api.* names in tests. 
Tests should use our own public APIs where they exist.
 2021-09-13 16:01:32 -04:00
Peter Hawkins
26e9ebcdae Move jax.api to jax._src.api. 
PiperOrigin-RevId: 368233837
 2021-04-13 09:43:24 -07:00
George Necula
aec6192ed9 Fix float0 handling 2021-01-06 14:15:42 +02:00
George Necula
1ca38f5d85 Implemented review feedback 
* removed comments
* raised the last exception as part of the CallbackException
 2021-01-05 10:51:32 +02:00
George Necula
64867ae808 [host_callback] Added host_callback.call to return values to the device 
See the PR for more details about this change
 2021-01-04 16:45:48 +02:00