ROCm/jax

Pull Requests Commits

skip test that fails due to known xla:cpu bug

froystig committed 5 years ago

db4a3eed

Merge pull request #4441 from jakevdp:choice

a-googler committed 5 years ago

5cfd0e9a

Implement jnp.choose

jakevdp committed 5 years ago

2fccbbf6

Merge pull request #4438 from hawkinsp:gelu

a-googler committed 5 years ago

95e3fcdf

Merge pull request #4439 from hawkinsp:selectandscatter

a-googler committed 5 years ago

113c8060

Merge pull request #4429 from apaszke:all_to_all_vmap_split

a-googler committed 5 years ago

d25f4041

Change select_and_scatter_add's batching rule to avoid unrolling.

hawkinsp committed 5 years ago

2d74037d

Adds an approximate=... keyword argument to jax.nn.gelu to select between the approximate and exact formulations of gelu.

hawkinsp committed 5 years ago

9b3bbe83

Merge pull request #4413 from alexminnaar:master

a-googler committed 5 years ago

b609040c

Merge pull request #4434 from google:numpy-dispatch-test-count

a-googler committed 5 years ago

9cf61593

histogramdd implemenation

alexminnaar committed 5 years ago

0893b08a

reduce test-case count of the numpy-dispatch CI check, to match our other x64-mode CI check

froystig committed 5 years ago

fd39901a

Merge pull request #4431 from tomhennigan:changelist/334843453

a-googler committed 5 years ago

93fbae62

Remove scalar float0 XLA representation.

tomhennigan committed 5 years ago

f18b9143

Merge pull request #4419 from rsepassi:master

a-googler committed 5 years ago

9a8ee95c

Merge pull request #4430 from LenaMartens:changelist/334791342

a-googler committed 5 years ago

a622220f

Fix dtype behavior with float0s in CustomVJP.

LenaMartens committed 5 years ago

cc0114a0

Merge pull request #4039 from LenaMartens:changelist/325216264

a-googler committed 5 years ago

69fda9ec

Merge pull request #4415 from jakevdp:indices-concrete

a-googler committed 5 years ago

768b0c1e

Call concrete_or_error on indices() argument

jakevdp committed 5 years ago

0b23a904

Merge pull request #4384 from jakevdp:cov-weights

a-googler committed 5 years ago

47da5121

Merge pull request #4425 from akihironitta:use-raise-from

a-googler committed 5 years ago

62f1e83e

Merge branch 'master' into use-raise-from

akihironitta committed 5 years ago

Verified d707ae17

Implement split_axis for all_to_all

apaszke committed 5 years ago

e61ca913

Add support for all_to_all over vmapped axes

apaszke committed 5 years ago

fa38f250

Merge pull request #4421 from jakevdp:multi-buf

a-googler committed 5 years ago

2d94a878

Merge pull request #4414 from n2cholas:master

a-googler committed 5 years ago

6b5a265b

Allow jax objects to be represented by multiple buffers

jakevdp committed 5 years ago

40016cc4

Use `raise from`

akihironitta committed 5 years ago

Verified 06170da6

Add --exclude_test_targets to test_util

Ryan Sepassi committed 5 years ago

97592c86

Older