Skip to main content

Browse the archive

https://github.com/google/jax
22 May 2024, 15:31:03 UTC
sort by:
Revision Author Date Message Commit Date
a0b92e0 Matthew Johnson 31 August 2019, 16:07:42 UTC experiments with a fixed_point implementation 18 September 2019, 01:25:43 UTC
480f461 Matthew Johnson 18 September 2019, 01:00:48 UTC Merge pull request #1255 from dsweaver/master Pass check_vjp, add VJP grad and Jac, use fun(y0, t, *args) syntax 18 September 2019, 01:00:48 UTC
981109c Matthew Johnson 18 September 2019, 00:44:27 UTC Merge pull request #1358 from nkitaev/issue1330 Reduce memory usage for argmax (fixes #1330) 18 September 2019, 00:44:27 UTC
145b2c3 Matthew Johnson 17 September 2019, 15:34:01 UTC bump version number for pypi 17 September 2019, 15:34:01 UTC
ac53afa Matthew Johnson 17 September 2019, 00:29:40 UTC Merge pull request #1360 from google/automasking tweaks to simplify masked jaxprs, rnn test 17 September 2019, 00:29:40 UTC
99b9e48 Matthew Johnson 16 September 2019, 23:30:42 UTC python2 fix for ShapeExpr slicing 16 September 2019, 23:30:42 UTC
6662da8 Matthew Johnson 16 September 2019, 22:47:43 UTC tweaks to simplify masked jaxprs, rnn test 16 September 2019, 22:47:43 UTC
13b1bca Matthew Johnson 16 September 2019, 21:49:31 UTC prevent device_put from being staged into jaxprs 16 September 2019, 21:49:31 UTC
0e7ea7e Nikita Kitaev 16 September 2019, 21:30:28 UTC Reduce memory usage for argmax (fixes #1330) 16 September 2019, 21:30:28 UTC
4d7f41b Peter Hawkins 16 September 2019, 20:57:50 UTC Merge pull request #1357 from hawkinsp/master Temporarily remove `jit` decorator on gather/scatter ops. 16 September 2019, 20:57:50 UTC
45a02f3 Peter Hawkins 16 September 2019, 20:57:07 UTC Temporarily remove `jit` decorator on gather/scatter ops. 16 September 2019, 20:57:07 UTC
d691f81 Peter Hawkins 16 September 2019, 16:21:38 UTC Merge pull request #1356 from hawkinsp/master Don't issue NaNs for LU decomposition of singular matrices. 16 September 2019, 16:21:38 UTC
b54f809 Peter Hawkins 16 September 2019, 15:45:10 UTC Don't issue NaNs for LU decomposition of singular matrices. 16 September 2019, 15:46:29 UTC
3d870fd Matthew Johnson 16 September 2019, 01:58:57 UTC Merge pull request #1354 from shoyer/shape-error Don't reuse SyntaxError in masking.py 16 September 2019, 01:58:57 UTC
4f5dcb0 Stephan Hoyer 15 September 2019, 21:51:14 UTC Don't reuse SyntaxError in masking.py This exception type is really intended for errors in *Python* syntax. In particular, IPython will add extra lines to the stack-trace to try to show the original line of Python where the error came from. Compare: In [2]: raise ValueError --------------------------------------------------------------------------- ValueError Traceback (most recent call last) <ipython-input-2-e4c8e09828d5> in <module> ----> 1 raise ValueError ValueError: In [3]: raise SyntaxError Traceback (most recent call last): File "/Users/shoyer/miniconda3/envs/jax-py37/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3326, in run_code exec(code_obj, self.user_global_ns, self.user_ns) File "<ipython-input-3-52c234f9b487>", line 1, in <module> raise SyntaxError File "<string>", line unknown SyntaxError So I think it's better just to define our own `ShapeSyntaxError`. 15 September 2019, 21:53:47 UTC
0bbf115 Matthew Johnson 15 September 2019, 21:29:36 UTC Merge pull request #1353 from google/automasking automasking updates 15 September 2019, 21:29:36 UTC
6e22b41 Matthew Johnson 15 September 2019, 21:11:15 UTC skip problematic tests on tpu 15 September 2019, 21:11:15 UTC
b71181d Matthew Johnson 13 September 2019, 23:30:22 UTC start writing nesting test 15 September 2019, 18:10:05 UTC
2832996 Matthew Johnson 13 September 2019, 21:36:33 UTC add a 'monomorphic dim' symbol, bug fixes 15 September 2019, 18:10:05 UTC
78c70ec Matthew Johnson 12 September 2019, 04:57:54 UTC add dynamic shape envs 15 September 2019, 18:10:04 UTC
d1531a8 Matthew Johnson 15 September 2019, 17:16:07 UTC Merge pull request #1352 from google/issue1350 fix broadcasting bug in rem jvp, fixes #1350 15 September 2019, 17:16:07 UTC
e945c9c Matthew Johnson 15 September 2019, 16:24:00 UTC add some 'manual' lax.rem autodiff tests 15 September 2019, 16:25:54 UTC
5b6b72c Matthew Johnson 15 September 2019, 15:45:58 UTC fix broadcasting bug in rem jvp, fixes #1350 15 September 2019, 15:45:58 UTC
b7b5328 Peter Hawkins 13 September 2019, 18:16:01 UTC Merge pull request #1346 from hawkinsp/master Add a DeviceArray._unstack() method that unpacks an array along its m… 13 September 2019, 18:16:01 UTC
a4793e1 Peter Hawkins 13 September 2019, 18:08:17 UTC Merge pull request #1343 from hawkinsp/gather JIT-compile index and index-update expressions. 13 September 2019, 18:08:17 UTC
ae329dc Peter Hawkins 13 September 2019, 17:44:11 UTC Add a DeviceArray._unstack() method that unpacks an array along its major dimension. Use it to implement pxla's shard_arg method for DeviceArrays; this is faster than slicing out each element one by one. 13 September 2019, 17:47:09 UTC
723456b Peter Hawkins 13 September 2019, 17:39:39 UTC Slice objects are unhashable, so unpack them into tuples for forming static arguments. 13 September 2019, 17:39:39 UTC
5ffddc1 Peter Hawkins 13 September 2019, 14:37:41 UTC JIT-compile index and index-update expressions. Improves the performance of indexing in op-by-op mode. 13 September 2019, 14:37:41 UTC
c9fcc0a Matthew Johnson 12 September 2019, 12:57:52 UTC Merge pull request #1337 from levskaya/fix_axis_index Make pxla.axis_index return signed indices. 12 September 2019, 12:57:52 UTC
4da08e9 Matthew Johnson 12 September 2019, 05:06:07 UTC Merge pull request #1340 from shoyer/remove-gufuncs Remove notebooks/gufuncs.ipynb 12 September 2019, 05:06:07 UTC
2b2da81 Stephan Hoyer 12 September 2019, 05:01:18 UTC Remove notebooks/gufuncs.ipynb It's now redundant with jax.experimental.vectorize, and I'd rather keep it in one place. 12 September 2019, 05:03:13 UTC
4ee28cf Anselm Levskaya 11 September 2019, 21:15:38 UTC Make pxla.axis_index return signed indices for type compatibility with other jax indices. 11 September 2019, 21:15:38 UTC
71d5a7a Matthew Johnson 11 September 2019, 16:29:26 UTC Merge pull request #1336 from google/fix-vmap-of-pmap fix vmap-of-pmap bug 11 September 2019, 16:29:26 UTC
37323c1 Matthew Johnson 11 September 2019, 13:22:25 UTC check output shape in testVmapOfPmap2 11 September 2019, 13:22:25 UTC
593a769 Matthew Johnson 11 September 2019, 13:17:56 UTC Merge pull request #1334 from levskaya/fix_xla_error_msg fix xla shape-checking error message 11 September 2019, 13:17:56 UTC
6f0e244 Matthew Johnson 11 September 2019, 13:01:32 UTC fix vmap-of-pmap bug thanks @romanngg and @inoryy 11 September 2019, 13:05:43 UTC
bdbb57e Peter Hawkins 11 September 2019, 13:00:30 UTC Merge pull request #1335 from hawkinsp/master Fix incorrect slogdet parity calculation in presence of batch dims 11 September 2019, 13:00:30 UTC
a5f67d5 Peter Hawkins 11 September 2019, 12:19:26 UTC Fix incorrect slogdet parity calculation in presence of batch dimensions. 11 September 2019, 12:19:26 UTC
53d4283 Anselm Levskaya 11 September 2019, 06:25:12 UTC fix xla shape-checking error message 11 September 2019, 06:25:12 UTC
705eb1c James Bradbury 10 September 2019, 21:49:17 UTC Merge pull request #1331 from google/jb/dot-general-batch Remove explicit broadcasts in vmap(dot_general) 10 September 2019, 21:49:17 UTC
b4b14b7 James Bradbury 10 September 2019, 20:58:23 UTC remove broadcasts from _dot_general_batch_rule 10 September 2019, 20:58:23 UTC
65f80d6 Matthew Johnson 10 September 2019, 14:56:11 UTC bump version number for pypi 10 September 2019, 14:56:11 UTC
ab34081 James Bradbury 10 September 2019, 03:30:38 UTC Merge pull request #1324 from romanngg/patch-1 Fix `check_nans` method 10 September 2019, 03:30:38 UTC
f51b6a3 Matthew Johnson 10 September 2019, 01:48:31 UTC Merge pull request #1329 from google/issue1316 make jvp only form JVPTracers with nonzero tangent 10 September 2019, 01:48:31 UTC
84992b5 Matthew Johnson 10 September 2019, 01:25:16 UTC add test for issue #1316 10 September 2019, 01:25:16 UTC
74f6269 Matthew Johnson 10 September 2019, 00:47:15 UTC make jvp only form JVPTracers with nonzero tangent fixes #1316 10 September 2019, 00:47:15 UTC
bcad02f Daniel Weaver 09 September 2019, 19:54:10 UTC Remove 64-bit mode for GPU performance 09 September 2019, 19:54:10 UTC
22afc8c Matthew Johnson 09 September 2019, 02:47:03 UTC Merge pull request #1326 from sschoenholz/master Tiny change to enable vmap of lax.reshape with dimension numbers. 09 September 2019, 02:47:03 UTC
6f2d22f Sam Schoenholz 08 September 2019, 21:19:10 UTC Tiny change to enable vmap with dimension numbers. 08 September 2019, 21:19:10 UTC
20f167d James Bradbury 07 September 2019, 06:50:19 UTC Merge pull request #1323 from google/ortho-typo fix typo in orthogonal init 07 September 2019, 06:50:19 UTC
302dcc7 Roman Novak 07 September 2019, 03:47:02 UTC Fix `check_nans` method 07 September 2019, 03:47:02 UTC
dbea4b6 Roman Novak 07 September 2019, 03:30:30 UTC Fix typos Hit these when trying to debug NaNs, appear to be just typos. 07 September 2019, 03:30:30 UTC
c4081d8 James Bradbury 06 September 2019, 23:13:04 UTC fix typo in orthogonal init 06 September 2019, 23:13:04 UTC
87f439d James Bradbury 06 September 2019, 21:44:26 UTC Merge pull request #1322 from google/jb/ortho-init Add orthogonal initializer 06 September 2019, 21:44:26 UTC
d233a56 James Bradbury 06 September 2019, 18:56:40 UTC Add orthogonal initializer 06 September 2019, 18:56:40 UTC
8f0355e Peter Hawkins 06 September 2019, 17:35:44 UTC Merge pull request #1319 from hawkinsp/master Fix compile error in cusolver.cc 06 September 2019, 17:35:44 UTC
c42444d Peter Hawkins 06 September 2019, 17:35:09 UTC Fix compile error in cusolver.cc 06 September 2019, 17:35:09 UTC
1ee2b75 Peter Hawkins 06 September 2019, 13:33:36 UTC Merge pull request #1314 from hawkinsp/svd Implement batched singular value decomposition. 06 September 2019, 13:33:36 UTC
701c34e Peter Hawkins 06 September 2019, 12:48:53 UTC Guard batched SVD tests by a jaxlib version check. 06 September 2019, 12:48:53 UTC
e087915 James Bradbury 06 September 2019, 00:45:13 UTC Merge pull request #1315 from Sohl-Dickstein/patch-1 improve numerical stability of softplus grads 06 September 2019, 00:45:13 UTC
c0c4aac Peter Hawkins 05 September 2019, 22:12:00 UTC Implement batched singular value decomposition. On GPU, switch to using the Jacobi implementation of SVD for matrices smaller than 32x32. The Jacobi implementation has an efficient implementation for batches of small matrices. 05 September 2019, 22:12:00 UTC
8f5cfef Jascha Sohl-Dickstein 05 September 2019, 22:10:48 UTC improve numerical stability of softplus grads 05 September 2019, 22:10:48 UTC
2b075f9 Skye Wanderman-Milne 05 September 2019, 21:48:02 UTC Update README to use jaxlib-0.1.28 05 September 2019, 21:48:02 UTC
2c6f74d Skye Wanderman-Milne 05 September 2019, 21:45:32 UTC Add CUDA 10.1 wheels. 05 September 2019, 21:45:32 UTC
5e0c0c9 Skye Wanderman-Milne 05 September 2019, 18:52:59 UTC Create shim to abstract over breaking changes between jaxlib 0.1.27 and 0.1.28. 05 September 2019, 20:51:13 UTC
b5130e1 James Bradbury 05 September 2019, 17:17:36 UTC Merge pull request #1307 from google/jb/rsqrt add primitive for rsqrt 05 September 2019, 17:17:36 UTC
eb7d594 Peter Hawkins 05 September 2019, 14:46:16 UTC Merge pull request #1309 from hawkinsp/block Change block_until_ready() to return `self` rather than nothing. 05 September 2019, 14:46:16 UTC
aa225a3 Peter Hawkins 05 September 2019, 14:27:37 UTC Merge pull request #1308 from hawkinsp/lu Implement jax.scipy.linalg.lu_solve. 05 September 2019, 14:27:37 UTC
612ffd0 Peter Hawkins 05 September 2019, 14:16:20 UTC Change block_until_ready() to return `self` rather than nothing. 05 September 2019, 14:16:20 UTC
454cb4f Peter Hawkins 05 September 2019, 13:59:47 UTC Implement jax.scipy.linalg.lu_solve. 05 September 2019, 13:59:47 UTC
73d512b Skye Wanderman-Milne 05 September 2019, 01:27:55 UTC Add nn/*.py to jax/BUILD 05 September 2019, 01:27:55 UTC
7dc95f1 Skye Wanderman-Milne 05 September 2019, 01:14:50 UTC Bump jaxlib version to 0.1.28 and update WORKSPACE. This pulls in breaking changes to the XLA client. 05 September 2019, 01:15:15 UTC
76156e5 Skye Wanderman-Milne 03 September 2019, 22:31:10 UTC Temporarily disable test_jit_device_assignment. Future breaking changes to jaxlib will break this test, so disable it until we update jaxlib and then can update jax and reenable the test. 04 September 2019, 23:25:04 UTC
35b63c7 James Bradbury 04 September 2019, 22:06:46 UTC add primitive for rsqrt 04 September 2019, 22:06:46 UTC
146b5d1 James Bradbury 04 September 2019, 21:48:20 UTC Merge pull request #1262 from google/jb/initializers Migrate initializers and activation functions to jax.nn 04 September 2019, 21:48:20 UTC
d90011b Peter Hawkins 04 September 2019, 20:58:18 UTC Merge pull request #1306 from hawkinsp/qr Use LAPACK and Cusolver to implement QR decomposition on CPU/GPU. 04 September 2019, 20:58:18 UTC
02426b3 Peter Hawkins 04 September 2019, 20:24:32 UTC Use LAPACK and Cusolver to implement QR decomposition on CPU/GPU. This should be faster; also adds support for complex QR decompositions. 04 September 2019, 20:24:32 UTC
315d3fe Matthew Johnson 04 September 2019, 15:54:41 UTC Merge pull request #1303 from google/definitely-not-automasking Automasking prototype 04 September 2019, 15:54:41 UTC
c3db5d7 Matthew Johnson 04 September 2019, 04:56:45 UTC fix dtype issue, python3 issue, sorting issue 04 September 2019, 14:16:47 UTC
bf28c44 James Bradbury 04 September 2019, 00:51:29 UTC address comments 04 September 2019, 00:51:29 UTC
96b8bb2 Matthew Johnson 04 September 2019, 00:18:23 UTC fix lax._canonicalize_shape for ShapeExprs 04 September 2019, 00:18:23 UTC
772fdb8 Matthew Johnson 04 September 2019, 00:09:27 UTC move automasking prototype into jax/interpreters Co-authored-by: Dougal Maclaurin <dougalm@google.com> 04 September 2019, 00:10:17 UTC
e12c8b0 Matthew Johnson 31 August 2019, 04:35:56 UTC experiments in import-time shape checking 04 September 2019, 00:10:17 UTC
f4d6591 Matthew Johnson 31 August 2019, 01:26:41 UTC revive dot example 04 September 2019, 00:10:17 UTC
2029927 Matthew Johnson 31 August 2019, 01:03:18 UTC concat is cool (packed not striped) 04 September 2019, 00:10:17 UTC
fbc85af Matthew Johnson 30 August 2019, 23:06:43 UTC made polymorphic jaxprs, reshape fail 04 September 2019, 00:10:17 UTC
a609ae7 Matthew Johnson 30 August 2019, 16:31:21 UTC set up a small shape language 04 September 2019, 00:10:17 UTC
976ff5f Matthew Johnson 29 August 2019, 01:39:54 UTC try out a ShapeExpr with polynomials 04 September 2019, 00:10:17 UTC
983b83f Matthew Johnson 28 August 2019, 14:09:58 UTC todo: test setup 04 September 2019, 00:10:17 UTC
cf637fd Matthew Johnson 28 August 2019, 14:05:55 UTC note next steps 04 September 2019, 00:10:17 UTC
75eb9b2 Matthew Johnson 28 August 2019, 04:43:56 UTC add part of a dot masking rule 04 September 2019, 00:10:17 UTC
ba82fe5 Matthew Johnson 28 August 2019, 02:04:22 UTC add some error checking 04 September 2019, 00:10:17 UTC
9bd5d37 Matthew Johnson 28 August 2019, 01:25:29 UTC split shape rules and masking rules 04 September 2019, 00:10:17 UTC
8a94e5c Matthew Johnson 28 August 2019, 00:42:50 UTC renames 04 September 2019, 00:10:17 UTC
bd86d41 Matthew Johnson 28 August 2019, 00:32:21 UTC add flattening, jit test 04 September 2019, 00:10:17 UTC
f5ee804 Matthew Johnson 28 August 2019, 00:09:55 UTC fix bug in scan masking rule 04 September 2019, 00:10:17 UTC
e254dc4 Matthew Johnson 26 August 2019, 19:34:57 UTC wip Co-authored-by: Dougal Maclaurin <dougalm@google.com> 04 September 2019, 00:10:17 UTC
4b7bc57 Matthew Johnson 23 August 2019, 18:21:37 UTC tweaks 04 September 2019, 00:10:17 UTC
5879dbc Matthew Johnson 23 August 2019, 17:28:27 UTC trying out ideas Co-authored-by: Dougal Maclaurin <dougalm@google.com> 04 September 2019, 00:10:17 UTC
5d3982e Skye Wanderman-Milne 03 September 2019, 22:19:32 UTC Update build_mac_jaxlib.yml 03 September 2019, 22:19:32 UTC
back to top