ROCm/jax

Pull Requests Commits

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

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

Fix the abstract eval and translation rule for all_to_all

apaszke committed 5 years ago

0d5f15f5

Merge pull request #4391 from apaszke:axis_index_handle_list

a-googler committed 5 years ago

c875ab3e

Merge pull request #4382 from hawkinsp:lu

a-googler committed 5 years ago

c9b3df3a

Merge pull request #4392 from SIben:fix_eigh_flakiness_cpu

a-googler committed 5 years ago

c6cd2f91

Newer Older