ROCm/jax

Pull Requests Commits

Revert "Allow JAX objects to be represented by multiple buffers"

hawkinsp committed 5 years ago

Verified a0e14b05

Merge pull request #3983 from jakevdp:device-put-tuple

a-googler committed 5 years ago

fa113388

Merge pull request #4342 from google:improve-tracer-error

a-googler committed 5 years ago

5b3cbc5e

mattjj committed 5 years ago

23a25dad

Merge pull request #4412 from jakevdp:axis-concrete

a-googler committed 5 years ago

aaa5724e

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

Use core.concrete_or_error() to improve errors in reductions

jakevdp committed 5 years ago

185590fc

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

Allow jax objects to be represented by multiple buffers

jakevdp committed 5 years ago

d1f80228

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

Older