ROCm/jax

Pull Requests Commits

mattjj committed 5 years ago

23a25dad

tweak error to mention *flattened* positions

mattjj committed 5 years ago

61fcea93

Merge branch 'master' into improve-tracer-error

mattjj committed 5 years ago

Verified f66c2eef

Merge pull request #4411 from google:omnistaging-doc-fix-links

a-googler committed 5 years ago

4d556837

fix links in omnistaging doc (hopefully...)

mattjj committed 5 years ago

be6e85a1

Merge pull request #4410 from google:add-omnistaging-doc

a-googler committed 5 years ago

5e194819

add author line, fix typo

mattjj committed 5 years ago

90b272aa

add omnistaging doc

mattjj committed 5 years ago

0fb12a21

Merge pull request #4368 from jakevdp:unop-check-array

a-googler committed 5 years ago

8d24d0af

Merge pull request #4406 from gnecula:print_debug

a-googler committed 5 years ago

95389e87

[host_callback] improved logging, and add instructions for debugging

gnecula committed 5 years ago

f46a4220

Merge pull request #4403 from SIben:eigh-edge-case

a-googler committed 5 years ago

ca32966c

Merge pull request #4402 from gnecula:omni_print

a-googler committed 5 years ago

a1da363c

[jax2tf] Fix casting in translation of eigh for empty arrays.

bchetioui committed 5 years ago

fd5294b0

Merge pull request #4170 from apaszke:all_to_all

a-googler committed 5 years ago

1bf80368

Allow vmapping all_to_all and implement a (slow) CPU and GPU translation

apaszke committed 5 years ago

acd4cc57

Tag tests that require omnistaging

gnecula committed 5 years ago

03970e97

[host_callback] Fixes and documentation changes for omnistaging

gnecula committed 5 years ago

2b615135

Merge pull request #4388 from apaszke:all_to_all_translation

a-googler committed 5 years ago

f7a09c63

Merge pull request #4364 from botev:api

a-googler committed 5 years ago

b8396c1d

Merge pull request #4387 from gnecula:print_doc

a-googler committed 5 years ago

aa85166e

Merge pull request #4400 from google:issue4389

a-googler committed 5 years ago

20c8c1ae

Merge pull request #4401 from google:update-pypi

a-googler committed 5 years ago

bdcefbd3

update version and changelog for pypi

mattjj committed 5 years ago

951bc756

add jax logo file

mattjj committed 5 years ago

ebf7c1b6

Merge pull request #4398 from google:lift-randomness-limit

a-googler committed 5 years ago

c7e0ef40

make_jaxpr return_shape use ShapeDtypeStruct, test

mattjj committed 5 years ago

d607164d

skip checks in big randomness test

mattjj committed 5 years ago

71f5f997

fix test for non-omnistaging

mattjj committed 5 years ago

96f5a3c4

remove limit on size of random arrays

mattjj committed 5 years ago

c42d736e

Older