ROCm/jax

Pull Requests Commits

improve scan error message test

mattjj committed 6 years ago

d01e6a96

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

mattjj committed 6 years ago

f3c1a5ca

Add "bazel shutdown" to the build script.

hawkinsp committed 6 years ago

8cbc475a

Merge pull request #742 from google/issue740

mattjj committed 6 years ago

Verified bf64d964

Merge pull request #741 from google/issue738

mattjj committed 6 years ago

Verified 52d73e43

handle tensordot with zero contracting dims

mattjj committed 6 years ago

dcb8584b

wrap np.trace axes (fixes #738)

mattjj committed 6 years ago

d0e1b7be

bump version number for pypi

mattjj committed 6 years ago

6539c8e4

Merge pull request #737 from google/issue121

mattjj committed 6 years ago

Verified 9f743a18

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

mattjj committed 6 years ago

f8aa563d

revise optimizers.py docstrings (no JaxTuples)

mattjj committed 6 years ago

adb15b7f

Merge pull request #736 from google/pytree-namedtuple

mattjj committed 6 years ago

Verified 72c03795

add test for namedtuple transparency

mattjj committed 6 years ago

ca66c769

make namedtuples transparent (act as pytree nodes)

mattjj committed 6 years ago

88f691f8

Merge pull request #735 from google/issue711

mattjj committed 6 years ago

Verified 5cbaf75d

fix bug in scan utility _convert_zeros

mattjj committed 6 years ago

4ab26774

Merge pull request #733 from google/issue732

mattjj committed 6 years ago

Verified 85881672

test cleanup per review

mattjj committed 6 years ago

0ac4e4b0

mention numpy promotion differences in README

mattjj committed 6 years ago

560b0ba9

change dtype promotion behavior for jit-invariance

mattjj committed 6 years ago

42a1ad43

catch pmap num_devices errors at compile time too

mattjj committed 6 years ago

bb5cbaab

Fix typo in BatchTrace.post_process_call

dougalm committed 6 years ago

7c348588

bump version number for pypi

mattjj committed 6 years ago

5afc3660

Merge pull request #730 from google/issue728

mattjj committed 6 years ago

Verified ed1ce34d

remove one lax parallel op from sphinx docs

mattjj committed 6 years ago

2f645dd3

Merge pull request #729 from google/pmap-improvements

mattjj committed 6 years ago

Verified 38750d33

add more tests that would have caught #728

mattjj committed 6 years ago

3b66c7ea

add test for issue #728

mattjj committed 6 years ago

b8207cab

BroadcastedIota needs integer type (fixes #728)

mattjj committed 6 years ago

8f9e4b12

add test for DeviceConstant repr

mattjj committed 6 years ago

b1fd8e6e

Older