ROCm/jax

Pull Requests Commits

mark lax traceables as boundary for filtered stack traces, and exclude lax frames as internal-only

froystig committed 5 years ago

3d9306b4

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

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

Merge pull request #4393 from hawkinsp:qr

a-googler committed 5 years ago

f3d61320

Enable tests for complex QR decomposition on TPU.

hawkinsp committed 5 years ago

89b98965

[jax2tf] Adjust tolerance in flaky float32 eigh test.

bchetioui committed 5 years ago

3360bee9

Consider lists as groups of axis names too

apaszke committed 5 years ago

2b7580c2

[host_callback] Update the documentation

gnecula committed 5 years ago

625be693

Enable fast TPU LU decomposition for complex types.

hawkinsp committed 5 years ago

7e1b826e

Merge pull request #4381 from jakevdp:check-arraylike

a-googler committed 5 years ago

80fa22cf

Merge pull request #4377 from apaszke:axis_index

a-googler committed 5 years ago

533fe28b

Merge pull request #4378 from apaszke:axis_index_nd

a-googler committed 5 years ago

99ffcc44

Call _check_arraylike in jax.numpy to improve error messages

jakevdp committed 5 years ago

ab273acb

Fix a faulty soft_pmap rule for axis_index

apaszke committed 5 years ago

8ac19c72

Older