Matthew Johnson
bf5a30cdbd
fix warning
2018-12-13 11:55:11 -08:00
Matthew Johnson
a285017110
fix failing tests (misc small bugs)
2018-12-13 11:52:41 -08:00
Peter Hawkins
3561b432c2
Add Cholesky, QR, and Triangular solve implementations.
...
* Adds lax.{cholesky,triangular_solve,qr}. Adds a JVP for Cholesky.
* Adds a transpose rule for add_p, needed by the Cholesky JVP.
* Adds np.linalg.{cholesky,qr,dot,matmul,trace}.
* Adds scipy.linalg.{cholesky,qr,solve_triangular,tril,triu}.
Pair programmed with mattjj.
2018-12-13 13:03:08 -05:00
Matthew Johnson
54bceee9e1
make num_generated_cases also settable by env var
2018-12-13 08:56:40 -08:00
Matthew Johnson
579665b1b0
add set_printoptions (same as onp version)
2018-12-13 07:28:59 -08:00
Peter Hawkins
22f9d50be0
Implement np.trace.
...
Also add code to forward np.{iinfo,finfo,issubdtype,integer} to their original numpy equivalents.
Fixes #29 .
2018-12-13 08:44:27 -05:00
Matthew Johnson
3fba60fcd5
Merge pull request #100 from vishwakftw/log2-log10-exp2-logaddexp-logaddexp2
...
Add log2, log10, logaddexp2, exp2 to lax_numpy
2018-12-13 05:09:39 -08:00
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
1f42d980b8
rename ResNet50Test -> ExamplesTest, remove some dots
2018-12-12 20:28:54 -08:00
Matthew Johnson
77d6fb4c01
transpose shouldn't transpose with identity perm
2018-12-12 20:28:54 -08:00
Matthew Johnson
538850e271
add misc numpy ops (c.f. #70 )
2018-12-12 20:28:54 -08:00
Peter Hawkins
9c63c0ff0a
Fix dtype semantics of numpy reductions to more closely match that of numpy.
...
Add tests for the dtype argument to reductions.
2018-12-12 21:53:18 -05:00
Peter Hawkins
7530f56809
Implement np.{diag,diagonal}.
2018-12-12 17:54:27 -05:00
Matthew Johnson
986514eeda
Merge pull request #71 from vishwakftw/scipy-stats-more-dists
...
Add a few popular distributions to jax.scipy.stats
2018-12-12 10:16:06 -08:00
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
vishwakftw
0e6a44f1d2
Add tests, fix small bugs in the logpdf functions
2018-12-12 20:41:48 +05:30
Dougal Maclaurin
9b835a180f
Added some docstrings to top-level transformations
2018-12-12 09:47:49 -05:00
Peter Hawkins
dad72f94da
Add missing license header to Python files.
2018-12-12 08:45:06 -05:00
Matthew Johnson
89349e5e6d
fix transpose issue in jacfwd and jacrev
2018-12-11 16:24:20 -08:00
Matthew Johnson
e788539e0a
add concatenate batching rule (c.f. #54 )
2018-12-11 16:11:54 -08:00
Matthew Johnson
97589a3d03
add batching rule for lax.pad (c.f. #54 )
2018-12-11 16:11:54 -08:00
Matthew Johnson
cbd9ace425
change vmap api to be curried ( closes #78 )
2018-12-11 12:52:09 -08:00
Matthew Johnson
9cd6027979
add python built-in complex type to array types
...
fixes #74
2018-12-11 12:14:57 -08:00
Roy Frostig
2b9a8bc13b
Merge pull request #77 from google/rot90
...
add rot90 and flip, adjust testOp test selection
2018-12-11 11:44:15 -08:00
Matthew Johnson
5a1aeca96c
add rot90 and flip, adjust testOp test selection
...
closes #55
2018-12-11 11:33:15 -08:00
Peter Hawkins
0f9a69fb81
Merge pull request #72 from google/unimplemented-funcs
...
Adding unimplemented functions more numpy modules
2018-12-11 13:58:07 -05:00
Matthew Johnson
7198e09465
enable skip_checks for merging to master
2018-12-11 13:22:07 -05:00
Matthew Johnson
ffecf2ccc5
fix bugs in zeros_like and complex transpose
2018-12-11 13:22:07 -05:00
Matthew Johnson
0d64aea6bb
clean up conv dimension_numbers handling
2018-12-11 13:22:07 -05:00
Dougal Maclaurin
1350db2b79
Added higher-order differentiation checks in lax_test and fixed some bugs. Conv tests currently failing.
2018-12-11 13:22:07 -05:00
Alex Wiltschko
0b7bed8778
Adding unimplemented functions to numpy.random, numpy.fft and numpy.linalg
2018-12-11 12:44:02 -05: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
7b9a3077f2
Reverting unnecessary blank space addition
2018-12-11 12:10:44 -05:00
vishwakftw
2d25a7430a
Add a few popular distributions to jax.scipy.stats
...
- Beta
- Exponential
- Gamma
- Laplace
- Uniform
2018-12-11 22:29:46 +05:30
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
Matthew Johnson
b907bcd5ae
improve jaxlib wheel building script
2018-12-09 08:59:08 -08:00
Matthew Johnson
499ea19e44
fix 'unreachable' bug in dot batching rule
2018-12-09 07:27:03 -08:00
Matthew Johnson
b7f6adaa90
add dot_general batching rule
2018-12-09 06:47:38 -08:00
Brett Koonce
70ef70b965
minor spelling tweaks
2018-12-08 15:20:50 -08:00
Matthew Johnson
ae36037abf
add ad_util.zeros_like translation rule
...
PAIR=hawkinsp
2018-12-08 09:35:24 -08:00
Matthew Johnson
10b61e08f7
fix symbolic zero handling in concat transpose
2018-12-08 06:01:25 -08:00
Dougal Maclaurin
30124b6da1
Added jit transformations to generated functions. Fixed bug in comparing numpy arrays for equality.
2018-12-08 00:03:34 -05:00
Roy Frostig
db748a046f
py3 compatibility fix
2018-12-07 10:00:14 -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
