ROCm/jax

Pull Requests Commits

loosen tolerance on testLuGrad

mattjj committed 7 years ago

6c47c9f2

fixed scan differentiation!

mattjj committed 7 years ago

096dfb9b

make check_grads check all orders <= given order

mattjj committed 7 years ago

1ecf7577

finer-grained passing and failing tests

mattjj committed 7 years ago

8c00df1d

Merge pull request #747 from shoyer/gufuncs-update

mattjj committed 7 years ago

Verified 11c871b9

add test showing bug in scan higher-order autodiff

mattjj committed 7 years ago

cf996730

Refresh notebooks/gufuncs.ipynb

shoyer committed 7 years ago

9fb5e953

Merge branch 'master' of github.com:google/jax

mattjj committed 7 years ago

6e99d9c0

update version number for pypi

mattjj committed 7 years ago

e8ef97fe

Merge pull request #745 from google/issue744

mattjj committed 7 years ago

Verified f6e394b5

improve scan error message test

mattjj committed 7 years ago

d01e6a96

improve scan error messages, add test (fixes #744)

mattjj committed 7 years ago

f3c1a5ca

Add "bazel shutdown" to the build script.

hawkinsp committed 7 years ago

8cbc475a

Merge pull request #742 from google/issue740

mattjj committed 7 years ago

Verified bf64d964

Merge pull request #741 from google/issue738

mattjj committed 7 years ago

Verified 52d73e43

handle tensordot with zero contracting dims

mattjj committed 7 years ago

dcb8584b

wrap np.trace axes (fixes #738)

mattjj committed 7 years ago

d0e1b7be

bump version number for pypi

mattjj committed 7 years ago

6539c8e4

Merge pull request #737 from google/issue121

mattjj committed 7 years ago

Verified 9f743a18

make jax.numpy.array(3) give 0D array, not scalar

mattjj committed 7 years ago

f8aa563d

revise optimizers.py docstrings (no JaxTuples)

mattjj committed 7 years ago

adb15b7f

Merge pull request #736 from google/pytree-namedtuple

mattjj committed 7 years ago

Verified 72c03795

add test for namedtuple transparency

mattjj committed 7 years ago

ca66c769

make namedtuples transparent (act as pytree nodes)

mattjj committed 7 years ago

88f691f8

Merge pull request #735 from google/issue711

mattjj committed 7 years ago

Verified 5cbaf75d

fix bug in scan utility _convert_zeros

mattjj committed 7 years ago

4ab26774

Merge pull request #733 from google/issue732

mattjj committed 7 years ago

Verified 85881672

test cleanup per review

mattjj committed 7 years ago

0ac4e4b0

mention numpy promotion differences in README

mattjj committed 7 years ago

560b0ba9

change dtype promotion behavior for jit-invariance

mattjj committed 7 years ago

42a1ad43

Newer Older