ROCm/jax

Pull Requests Commits

factor out ode tests into tests/ode_test.py

mattjj committed 6 years ago

217bdcf4

Merge pull request #1255 from dsweaver/master

mattjj committed 6 years ago

Verified 480f461e

Merge pull request #1358 from nkitaev/issue1330

mattjj committed 6 years ago

Verified 981109c6

bump version number for pypi

mattjj committed 6 years ago

145b2c31

Merge pull request #1360 from google/automasking

mattjj committed 6 years ago

Verified ac53afa5

python2 fix for ShapeExpr slicing

mattjj committed 6 years ago

99b9e485

tweaks to simplify masked jaxprs, rnn test

mattjj committed 6 years ago

6662da82

prevent device_put from being staged into jaxprs

mattjj committed 6 years ago

13b1bca9

Reduce memory usage for argmax (fixes #1330)

nkitaev committed 6 years ago

0e7ea7e3

Merge pull request #1357 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 4d7f41ba

Temporarily remove `jit` decorator on gather/scatter ops.

hawkinsp committed 6 years ago

45a02f39

Merge pull request #1356 from hawkinsp/master

hawkinsp committed 6 years ago

Verified d691f812

Don't issue NaNs for LU decomposition of singular matrices.

hawkinsp committed 6 years ago

b54f8099

Merge pull request #1354 from shoyer/shape-error

mattjj committed 6 years ago

Verified 3d870fdd

Don't reuse SyntaxError in masking.py

shoyer committed 6 years ago

4f5dcb00

Merge pull request #1353 from google/automasking

mattjj committed 6 years ago

Verified 0bbf1151

skip problematic tests on tpu

mattjj committed 6 years ago

6e22b418

start writing nesting test

mattjj committed 6 years ago

b71181d3

add a 'monomorphic dim' symbol, bug fixes

mattjj committed 6 years ago

28329964

add dynamic shape envs

mattjj committed 6 years ago

78c70ecd

Merge pull request #1352 from google/issue1350

mattjj committed 6 years ago

Verified d1531a88

add some 'manual' lax.rem autodiff tests

mattjj committed 6 years ago

e945c9c1

fix broadcasting bug in rem jvp, fixes #1350

mattjj committed 6 years ago

5b6b72c2

Merge pull request #1346 from hawkinsp/master

hawkinsp committed 6 years ago

Verified b7b53285

Merge pull request #1343 from hawkinsp/gather

hawkinsp committed 6 years ago

Verified a4793e1a

Add a DeviceArray._unstack() method that unpacks an array along its major dimension.

hawkinsp committed 6 years ago

ae329dcd

Slice objects are unhashable, so unpack them into tuples for forming static arguments.

hawkinsp committed 6 years ago

723456bc

JIT-compile index and index-update expressions.

hawkinsp committed 6 years ago

5ffddc18

Merge pull request #1337 from levskaya/fix_axis_index

mattjj committed 6 years ago

Verified c9fcc0af

Merge pull request #1340 from shoyer/remove-gufuncs

mattjj committed 6 years ago

Verified 4da08e95

Older