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

16376 Commits

Author SHA1 Message Date
He Jia
dc71e0d965
Update __init__.py 2019-01-19 22:03:18 +08:00
He Jia
7b6164c20a
add scipy.stats.uniform.pdf 2019-01-19 20:39:09 +08:00
He Jia
fc2de383bf
add scipy.stats.laplace.pdf 2019-01-19 20:38:07 +08:00
He Jia
beb2662604
add scipy.stats.gamma.pdf 2019-01-19 20:37:08 +08:00
He Jia
b2e4882c8a
add scipy.stats.expon.pdf 2019-01-19 20:36:18 +08:00
He Jia
8e311059e5
add scipy.stats.beta.pdf 2019-01-19 20:29:27 +08:00
He Jia
8e9d50136a
add scipy.stats.norm.pdf 2019-01-19 20:25:14 +08:00
Peter Hawkins
4ea3f2cd8c Add complex128 support to LAPACK bindings. 
Update XLA to incorporate 7a283b835b .
 2019-01-18 16:54:07 -05:00
Matthew Johnson
42bd253cbb add prng design doc in markdown format 2019-01-18 07:38:43 -08:00
Peter Hawkins
62d946123c Use complexfloating instead of complex to suppress NumPy warning. 
Fixes #255.
 2019-01-17 13:41:40 -05:00
Jamie Townsend
38b2da82c5 Use slots for tracer classes 2019-01-16 16:51:54 +00:00
Peter Hawkins
86d8915c3d Add Sphinx-generated reference documentation for JAX. 2019-01-16 09:13:31 -05:00
Peter Hawkins
4792b9bed3
Merge pull request #231 from hawkinsp/complex 
Add preliminary support for np.complex128.
 2019-01-15 11:44:27 -05:00
Peter Hawkins
c31b465259
Merge pull request #237 from hawkinsp/gather_scatter 
Add lax.gather and lax.scatter_add.
 2019-01-15 11:20:52 -05:00
Peter Hawkins
2b383bdbd9 Increase cache size to 4096. 2019-01-15 10:32:58 -05:00
Peter Hawkins
3266bb3122 Change linear_util.memoize to use an LRU cache. 
Add util.OrderedDict that retrofits a move_to_end method onto Python 2 OrderedDicts.
 2019-01-14 21:48:28 -05:00
Peter Hawkins
05b1049e49 Change util.memoize to be an LRU cache with a default size of 64 entries. 
The goal is to limit peak memory when running a large number of computations, e.g., the test suite.
 2019-01-14 20:11:08 -05:00
Matthew Johnson
ade92703ec guard np.take_along_axis test in case of old numpy 2019-01-14 14:12:09 -08:00
Peter Hawkins
a21c3c4562 Fix definition of rand_uniform(). 2019-01-14 16:48:07 -05:00
Peter Hawkins
e00dc5d39d Restrict the range of np.tan() test to [-1.5, 1.5) to avoid numerical problems. 2019-01-14 15:28:53 -05:00
Colin
33bd020bd5 Use repr of value for DeviceArray.__repr__ 
Currently, DevieArray.__repr__ returns the class name and the size of
the object. This is inconsistent with np.ndarray.__repr__ which produces
code to reconstruct the array (including the contents, unless the
contents are too long to print cleanly). This provides a strange
reminder that we aren't using "real" numpy so it may be nicer to just
use the standard numpy __repr__ instead.
 2019-01-14 12:01:14 -08:00
Peter Hawkins
5fac477a8a Fix bug in scatter transpose rule. 
Add some simple gather and scatter tests.
 2019-01-14 14:33:40 -05:00
Peter Hawkins
39257b2442 Work on scatter JVP/transpose. 2019-01-14 10:28:35 -05:00
Peter Hawkins
9812bea1ef Merge remote-tracking branch 'google/master' into gather_scatter 2019-01-14 08:24:01 -05:00
Matthew Johnson
f7c6284313 add sort_along_axis, no tests (c.f. #220) 2019-01-13 12:17:08 -08:00
Matthew Johnson
30683f8f43 add numpy.argsort (c.f. #221) 2019-01-13 11:56:36 -08:00
Matthew Johnson
54886bd310 add sort_key_val batching rule (fixes #221) 2019-01-13 11:56:36 -08:00
Matthew Johnson
5a4713f108 add tests for np.sort (c.f. #221) 2019-01-13 11:56:36 -08:00
Matthew Johnson
70d1a00443 set behavior when random.randint has invalid range 
(closes #222)
 2019-01-12 13:12:40 -08:00
Matthew Johnson
6be53c7a0b fix isinstance check in indexing (fixes #227) 2019-01-12 11:09:15 -08:00
Peter Hawkins
d43c65dcd8 Add preliminary support for np.complex128. 
Only lightly tested.
 2019-01-11 18:22:43 -05:00
Peter Hawkins
8cb73c1c2e
Merge pull request #225 from hawkinsp/complex 
Test more Numpy ops for complex types.
 2019-01-11 16:01:52 -05:00
Peter Hawkins
4a8f088ac9 Simplify and fix test failures of np.clip for cases where one of the inputs was None. 2019-01-11 15:50:51 -05:00
Peter Hawkins
65efd45fc9 Test more Numpy ops for complex types. 
Fix a number of ops that did not handle complex numbers the same way as regular numpy.
 2019-01-11 14:49:42 -05:00
vishwakftw
e21cb1a99f Add complex support for SVD 2019-01-10 23:13:56 +05:30
Peter Hawkins
087d0730c5
Merge pull request #213 from hawkinsp/pad 
Implement the `mode='constant'` case of `np.pad`.
 2019-01-10 08:57:09 -05:00
Peter Hawkins
0fa5af9dbb Implement the mode='constant' case of np.pad. 2019-01-09 21:26:22 -05:00
Peter Hawkins
9e91fb42cb Make LU JVP work for rectangular matrices. 
Fix bug in BLAS TRSM kernel if left_side=False.
 2019-01-09 15:52:34 -05:00
Peter Hawkins
bf4c60635b Make JVP work for square matrices. 2019-01-09 15:32:23 -05:00
Peter Hawkins
525a492d5d Start work on a JVP rule for LU decomposition. 2019-01-09 15:32:23 -05:00
Peter Hawkins
adaa344400 More progress on scatter/gather. 2019-01-09 10:58:44 -05:00
Peter Hawkins
5e48420f12 First attempt at lax.gather and lax.scatter. 2019-01-08 21:34:48 -05:00
Peter Hawkins
04bfd30fd1
Merge pull request #197 from vishwakftw/svd 
Add SVD to jax/numpy/linalg, jax/scipy/linalg
 2019-01-08 13:26:44 -05:00
vishwakftw
c509d48b08 Address comments 
- Fix acronym
- Include test for compute_uv=False
- Remove unnecessary imports
 2019-01-08 21:48:15 +05:30
Peter Hawkins
1643b1ecc8 Implement np.square. 2019-01-08 09:26:11 -05:00
vishwakftw
e16f31e94f Merge branch 'master' of https://github.com/google/jax into svd 2019-01-08 09:30:05 +05:30
vishwakftw
954b047bea Address review comments 2019-01-08 09:24:48 +05:30
Matthew Johnson
6687734bb8
Merge pull request #208 from google/unpack-full-lower 
make tuple unpacking cause a full_lower
 2019-01-07 17:50:36 -08:00
Matthew Johnson
685130b766
Merge pull request #205 from google/issue204 
make lax.full require concrete (static) shapes
 2019-01-07 17:28:11 -08:00
Matthew Johnson
280b3fe2fc python3 likes list(map(...)) 2019-01-07 16:54:51 -08:00