ROCm/jax

Pull Requests Commits

refactor api wrapping code

mattjj committed 7 years ago

affd558c

add generic lower_fun for xla

mattjj committed 7 years ago

9503faaf

add the (temporary) files we were playing with

mattjj committed 7 years ago

ca8e0e5b

explorations with @duvenaud and @dougalm

mattjj committed 7 years ago

e17672bd

Merge pull request #179 from google/prng-key-vmap

mattjj committed 7 years ago

Verified d2bfc9ad

add batching test for PRNGKeys and random sampling

mattjj committed 7 years ago

4ed5df85

replace PRNGKey class with uint32[2] array

mattjj committed 7 years ago

072e6f78

Merge pull request #178 from google/dkd

mattjj committed 7 years ago

Verified c5c6e6c5

only test np.polyval on nonscalar array shapes

mattjj committed 7 years ago

5d2fc7c0

add np.append and np.polyval

mattjj committed 7 years ago

9c49a9bf

Merge pull request #175 from google/stax-fan-in-concat

mattjj committed 7 years ago

Verified 61d5d79e

add stax.FanInConcat (fixes #174)

mattjj committed 7 years ago

b4246163

add a placeholder comment for lbr tests

froystig committed 7 years ago

b2d6ce17

skip linalg QR test on cpu

froystig committed 7 years ago

0211934c

bump version number for pypi

mattjj committed 7 years ago

95a86cd0

add error message if Dropout gets no rng key

mattjj committed 7 years ago

59142a24

Merge pull request #169 from google/boolean-indexing-support

mattjj committed 7 years ago

Verified 041ef322

loosen scipy test tol to fix benign failure

mattjj committed 7 years ago

b900339b

add non-advanced boolean indexing support

mattjj committed 7 years ago

6d6b5263

Merge pull request #168 from google/vmap-for-dynamic-slice

mattjj committed 7 years ago

Verified fd3645ab

add batching (vmap) rule for lax.dynamic_slice

mattjj committed 7 years ago

d48e7ef4

Merge pull request #167 from hawkinsp/master

hawkinsp committed 7 years ago

Verified 648fd436

skip numerically unstable test

froystig committed 7 years ago

87824ee0

Fix missing `raise` in error path.

hawkinsp committed 7 years ago

b1ff7b4f

Implement numpy.linalg.solve and scipy.linalg.solve.

hawkinsp committed 7 years ago

06135fa6

Merge pull request #163 from hawkinsp/master

hawkinsp committed 7 years ago

Verified 54772e1b

Fix test failures due to type mismatches in linear algebra tests.

hawkinsp committed 7 years ago

a4386457

Merge pull request #164 from google/revert-155-tfds

alexbw committed 7 years ago

Verified df59c512

Revert "Add TensorFlow Datasets data-loading notebook"

alexbw committed 7 years ago

Verified b1f91685

Use get() rather than a try-catch block in memoized function lookup.

hawkinsp committed 7 years ago

8127392a

Older