mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 504 Branches 263 Tags
rocm-main
Commit Graph

26443 Commits

Author SHA1 Message Date
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
38c7a07248 download mnist example data w/ six.moves.request 
fixes
2018-12-09 10:54:37 -08:00
Roy Frostig
269de6010d
Merge pull request from craffel/master 
Require protobuf 3.6.0 or later
 2018-12-09 13:16:43 -05:00
Colin
2bbc046d9d Require protobuf 3.6.0 or later 2018-12-09 10:11:21 -08:00
Matthew Johnson
b907bcd5ae improve jaxlib wheel building script 2018-12-09 08:59:08 -08:00
Matthew Johnson
f50996df64 update gufunc notebook w/ dot_general batch rule 2018-12-09 07:46:24 -08:00
Matthew Johnson
4d6e7c8900 bump version number for pypi 2018-12-09 07:45:12 -08:00
Matthew Johnson
499ea19e44 fix 'unreachable' bug in dot batching rule 2018-12-09 07:27:03 -08:00
Matthew Johnson
1469e82361 bump version number for pypi 2018-12-09 06:53:22 -08:00
Matthew Johnson
af6891399f
Merge pull request from google/dot-general-batching-rule 
add dot_general batching rule
 2018-12-09 09:48:30 -05:00
Matthew Johnson
b7f6adaa90 add dot_general batching rule 2018-12-09 06:47:38 -08:00
Matthew Johnson
5a604d5b14
add more contributors to readme 2018-12-09 01:24:38 -05:00
Peter Hawkins
ae4d06f564
Merge pull request from brettkoonce/minor_sp 
minor spelling tweaks
 2018-12-08 18:23:59 -05:00
Brett Koonce
70ef70b965 minor spelling tweaks 2018-12-08 15:20:50 -08:00
Peter Hawkins
f2f661366f
Merge pull request from google/nnet-notebook 
Bare-bones neural network and data loading example notebook
 2018-12-08 13:56:51 -05:00
Alex Wiltschko
f5e6e88b89 Delete unused code in data loader, new architecture, some cleanups 2018-12-08 12:44:10 -05:00
Matthew Johnson
ae36037abf add ad_util.zeros_like translation rule 
PAIR=hawkinsp
 2018-12-08 09:35:24 -08:00
Alex Wiltschko
e22f722e28 Fixes for neural network notebook under review 2018-12-08 12:35:06 -05:00
Dougal Maclaurin
9f61472a91
Update README.md 2018-12-08 11:20:40 -05:00
Dougal Maclaurin
0cd84efac1
Small changes to readme 2018-12-08 11:19:03 -05:00
Matthew Johnson
31d9443e49
Merge pull request from google/concatenate-transpose-rule-bug 
fix symbolic zero handling in concat transpose
 2018-12-08 09:33:52 -05:00
Alex Wiltschko
13d8cb1e7d Bare-bones neural network and data loading example notebook 2018-12-08 09:23:29 -05:00
Matthew Johnson
10b61e08f7 fix symbolic zero handling in concat transpose 2018-12-08 06:01:25 -08:00
Matthew Johnson
ffd6951719
Merge pull request from google/center-logo 
attempt to center-justify the jax logo in readme
 2018-12-08 08:05:53 -05:00
Matthew Johnson
9d484b6d4c
attempt to center-justify the jax logo in readme 2018-12-08 08:05:16 -05:00
Matthew Johnson
9e739fbfc5
Merge pull request from dsblank/patch-1 
Typo, Python parens
 2018-12-08 07:48:52 -05:00
Doug Blank
13163ffefe
Typo, Python parens 2018-12-08 03:40:33 -05:00
Matthew Johnson
d2b0e30d54
update readme (accidentally lost a name!) 2018-12-08 00:56:25 -05: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
Matthew Johnson
0900db35b6
Add ToC to readme 2018-12-07 21:51:36 -05:00
Peter Hawkins
0528192bd9
Remove Mac OS X from the list of CUDA pip packages 
We haven't created this pip package.
 2018-12-07 21:09:09 -05:00
Matthew Johnson
70604f4309 add scipy.stats to setup.py (should use find_packages()) 2018-12-07 15:05:20 -08:00
Stephan Hoyer
5516019656 Fix link in gufuncs notebook 2018-12-07 16:30:30 -05:00
Dougal Maclaurin
69b0c2299c
Merge pull request from MikeInnes/patch-1 
Typo
 2018-12-07 16:21:01 -05:00
Mike J Innes
0505ce2950
typo 2018-12-07 16:16:37 -05:00
Matthew Johnson
5bbf3ee077
Merge pull request from shoyer/gufuncs-example 
Notebook showing how to write gufuncs with vmap
 2018-12-07 15:52:01 -05:00
Stephan Hoyer
89ec22e5bd Fixes per review 2018-12-07 12:28:41 -08:00
Stephan Hoyer
d6e2dce9f6 Notebook showing how to write gufuncs with vmap 2018-12-07 12:07:29 -08:00
Dougal Maclaurin
7be7d42484 Fixed json bug 2018-12-07 14:46:09 -05:00
Dougal Maclaurin
7523975c2e Updated installs in notebook 2018-12-07 14:44:40 -05:00
Matthew Johnson
13ed022245
Merge pull request from google/readme 
Populate readme
 2018-12-07 14:25:41 -05:00
Matthew Johnson
16658b97c4 fix typo in wheel path 2018-12-07 11:23:40 -08:00
Dougal Maclaurin
59fc69e0d4 Small edits to readme 2018-12-07 11:16:51 -08:00
Dougal Maclaurin
838d9e12fe Removed go/jax link and updated wording 2018-12-07 11:16:51 -08:00
Matthew Johnson
948a8db0ad populate readme with ill content 2018-12-07 11:16:51 -08:00
Matthew Johnson
ee1728c2af bump version for pypi 2018-12-07 11:15:50 -08:00
Roy Frostig
e5d89ce908 update call to choice 2018-12-07 10:51:02 -08:00
Roy Frostig
6c2cdfc318 py3 compatibility: list of keys 2018-12-07 10:45:48 -08:00
Roy Frostig
fe0a26936c use custom choice function in generated tests 2018-12-07 10:40:21 -08:00
Roy Frostig
479f256abd py3 iterator compatibility 2018-12-07 10:22:15 -08:00