vishwakftw
52d329c621
Add log2, log10, logaddexp2, exp2 to lax_numpy
...
I also took this opportunity to add some missing decorators
2018-12-13 12:57:49 +05:30
Matthew Johnson
af8223a2ad
Merge pull request #90 from hawkinsp/master
...
Add implementations of np.{tri,tril,triu}.
2018-12-12 09:17:40 -08:00
Peter Hawkins
3c11d4f780
Add implementations of np.{tri,tril,triu}.
2018-12-12 12:05:49 -05:00
Matthew Johnson
d3ec0b23c5
fix miscellaneous bugs, incl. complex abs grad
2018-12-12 12:03:57 -05:00
Matthew Johnson
5a1aeca96c
add rot90 and flip, adjust testOp test selection
...
closes #55
2018-12-11 11:33:15 -08:00
Roy Frostig
887c500368
Merge pull request #69 from google/unimplemented-funcs
...
Automated detection of unimplemented functions
2018-12-11 09:21:57 -08:00
Alex Wiltschko
6873760f4c
Remove unused lines
2018-12-11 11:56:03 -05:00
Alex Wiltschko
763f860135
Automated detection of unimplemented functions
2018-12-11 11:52:31 -05:00
Matt Graham
7180eb031b
More informative error on trying to concatenate 0-dim arrays.
2018-12-11 13:58:33 +08:00
Peter Hawkins
38927153b1
Fix support for arrays with size-0 dimensions.
...
* Fix broadcasting rules to support size-0 dimensions.
* Add tests for size-0 dimensions. This required extending the test harness to support testing shapes that aren't necessarily broadcast compatible.
* Fix test utils to support size-0 dimensions.
2018-12-10 09:00:22 -05:00
Brett Koonce
70ef70b965
minor spelling tweaks
2018-12-08 15:20:50 -08:00
Peter Hawkins
95eb58f448
Implement np.repeat() for scalar repeats.
2018-12-07 12:03:22 -05:00
Dougal Maclaurin
d89c7c2457
Merge branch 'master' into quickish-check
2018-12-07 11:13:41 -05:00
Matthew Johnson
18920337f3
add eye in lax_numpy
2018-12-07 07:34:52 -08: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
Dougal Maclaurin
8a3241e36b
Merge branch 'master' into quickish-check
2018-12-06 19:02:28 -05:00
Dougal Maclaurin
494c2dbae2
Merge branch 'master' into quickish-check
2018-12-06 18:58:26 -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
Peter Hawkins
f2795cbdea
[JAX] Add a NUMPY_SCALAR_SHAPE constant shape to test_utils.py to allow tests for both numpy scalars as distinct from 0D ndarrays.
...
Fix mishandling of scalars when passed to np.reshape().
PiperOrigin-RevId: 224326107
2018-12-06 06:47:58 -08:00
Dougal
d7b7200884
Error message for unimplemented numpy functions
2018-12-05 15:55:01 -05:00
Matthew Johnson
2597300c7f
source sync
...
PiperOrigin-RevId: 224157599
2018-12-05 09:36:16 -08:00
Dougal Maclaurin
ca2634ea5d
source sync
...
PiperOrigin-RevId: 222923229
2018-11-27 16:51:22 -08:00
Peter Hawkins
e180f08113
source sync
...
PiperOrigin-RevId: 222451919
2018-11-21 20:22:51 -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
