ROCm/jax

Pull Requests Commits

Merge pull request #237 from hawkinsp/gather_scatter

hawkinsp committed 6 years ago

Verified c31b4652

Merge pull request #243 from hawkinsp/lru

hawkinsp committed 6 years ago

Verified 2b1d0fb8

Increase cache size to 4096.

hawkinsp committed 6 years ago

2b383bdb

Change linear_util.memoize to use an LRU cache.

hawkinsp committed 6 years ago

3266bb31

Change util.memoize to be an LRU cache with a default size of 64 entries.

hawkinsp committed 6 years ago

05b1049e

Merge pull request #242 from hawkinsp/testfix

hawkinsp committed 6 years ago

Verified 3f63684b

Merge pull request #241 from google/take-along-axis-with-old-numpy

mattjj committed 6 years ago

Verified 0d8255f2

Fix incorrect norm definition in SVD test case, and adjust test tolerances.

hawkinsp committed 6 years ago

bc7231e6

guard np.take_along_axis test in case of old numpy

mattjj committed 6 years ago

ade92703

Merge pull request #238 from craffel/DeviceArray__repr__

mattjj committed 6 years ago

Verified 7404cb95

Relax tolerance for scatter/gather gradient tests.

hawkinsp committed 6 years ago

f16de58b

Fix definition of rand_uniform().

hawkinsp committed 6 years ago

a21c3c45

Merge pull request #240 from hawkinsp/numpyfix

hawkinsp committed 6 years ago

Verified d117dfbc

Work around breakage due to Numpy 1.16.0 version update.

hawkinsp committed 6 years ago

6e684fd3

Restrict the range of np.tan() test to [-1.5, 1.5) to avoid numerical problems.

hawkinsp committed 6 years ago

e00dc5d3

Use repr of value for DeviceArray.__repr__

craffel committed 6 years ago

33bd020b

Fix bug in scatter transpose rule.

hawkinsp committed 6 years ago

5fac477a

Work on scatter JVP/transpose.

hawkinsp committed 6 years ago

39257b24

Merge remote-tracking branch 'google/master' into gather_scatter

hawkinsp committed 6 years ago

9812bea1

Merge pull request #236 from google/numpy-take-along-axis

mattjj committed 6 years ago

Verified fc170b2b

add tests for numpy.take_along_axis (fixes #220)

mattjj committed 6 years ago

4a1d189d

add sort_along_axis, no tests (c.f. #220)

mattjj committed 6 years ago

f7c62843

Merge pull request #234 from google/issue221

mattjj committed 6 years ago

Verified 1ea77af3

add numpy.argsort (c.f. #221)

mattjj committed 6 years ago

30683f8f

add sort_key_val batching rule (fixes #221)

mattjj committed 6 years ago

54886bd3

add tests for np.sort (c.f. #221)

mattjj committed 6 years ago

5a4713f1

Merge pull request #235 from google/fix-failing-linalg-tests

mattjj committed 6 years ago

Verified dfa2cb82

double linalg test tolerances to avoid flakiness

mattjj committed 6 years ago

78dbc031

Update Jaxlib versions in Colab notebooks.

hawkinsp committed 6 years ago

61504a63

Rename build/BUILD to build/BUILD.bazel.

hawkinsp committed 6 years ago

0316d314

Newer Older