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

57 Commits

Author SHA1 Message Date
James Bradbury
ce34cb73c7 fix orthogonal initializer and reenable test 2019-10-29 11:34:20 -07:00
Peter Hawkins
0dd720cd8a
Disable some tests that fail. (#1587) 
Add a BUILD rule for experimental/vectorize.py.
 2019-10-29 11:04:55 -04:00
root
43be8d8ef8 Fix NaNs in grad(jax.nn.elu) for large inputs. 2019-10-21 11:48:58 +00:00
James Bradbury
23f06b417f add initializer tests 2019-10-03 12:01:21 -07:00
James Bradbury
b82673dcdb add check_dtypes 2019-09-27 12:11:18 -04:00
James Bradbury
fa0a684af6 address comments 2019-09-21 10:11:53 -04:00
James Bradbury
9ef53120d2 add regression tests 2019-09-21 01:04:26 -04:00