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

11 Commits

Author SHA1 Message Date
Matthew Johnson
a285017110 fix failing tests (misc small bugs) 2018-12-13 11:52:41 -08:00
Alex Wiltschko
0b7bed8778 Adding unimplemented functions to numpy.random, numpy.fft and numpy.linalg 2018-12-11 12:44:02 -05:00
Matthew Johnson
bbc92ce6eb
Split out jax and jaxlib packages (#11) 
factor out 'jaxlib' as separate package
 2018-12-06 21:35:03 -05:00
Peter Hawkins
c1b9eb19ea [JAX] Change semantics of dtype promotion to just call numpy.result_type. 
* Enable tests for numpy scalars in lax_numpy_test.py.
* Fix invalid promotion in random.py.
* Split tests for bitwise ops into their own test case and test mixed signedness.
* Add complex64 to the set of types supported by abstractify.
 2018-12-06 13:25:42 -05:00
Matthew Johnson
c293f7c875 minor: add @jit to threefry hash function in random.py 
PiperOrigin-RevId: 222841601
 2018-11-27 16:51:13 -08:00
Matthew Johnson
8317cc3618 source sync 
PiperOrigin-RevId: 222484671
 2018-11-21 20:32:33 -08:00
Matthew Johnson
2ae9a2bc35 source sync 
PiperOrigin-RevId: 222461242
 2018-11-21 20:32:16 -08:00
Peter Hawkins
e180f08113 source sync 
PiperOrigin-RevId: 222451919
 2018-11-21 20:22:51 -08:00
Peter Hawkins
065bb0baa2 source sync 
PiperOrigin-RevId: 222291726
 2018-11-21 20:22:41 -08:00
Matthew Johnson
46c6a9170f sync updates 2018-11-19 07:47:59 -08:00
Matthew Johnson
a30e858e59 populating source tree 2018-11-17 18:03:33 -08:00