ROCm/jax

Pull Requests Commits

equation context for undefined var reads in jaxpr typechecker

froystig committed 5 years ago

e7979258

Merge pull request #4456 from jakevdp:small-fix

a-googler committed 5 years ago

91755e5c

Merge pull request #4455 from johnpjf:patch-2

a-googler committed 5 years ago

9273e8c4

Merge pull request #4444 from jakevdp:dtype-check

a-googler committed 5 years ago

efeed6c1

remove redundant if-else

jakevdp committed 5 years ago

7854a235

Update Common_Gotchas_in_JAX.ipynb

johnpjf committed 5 years ago

Verified b357005e

Merge pull request #4442 from hawkinsp:docs

a-googler committed 5 years ago

d388b626

Merge pull request #4446 from n2cholas:master

a-googler committed 5 years ago

51538bd7

Update examples/differentially_private_sgd.py

n2cholas committed 5 years ago

Verified 238b4a12

Merge pull request #4443 from jakevdp:concrete

a-googler committed 5 years ago

0e92124c

fix dp sgd example

n2cholas committed 5 years ago

2ced1635

Merge pull request #4445 from google:lax-ad-test-skip

a-googler committed 5 years ago

6d5ace28

skip test that fails due to known xla:cpu bug

froystig committed 5 years ago

db4a3eed

Call check_user_dtype on all user dtypes

jakevdp committed 5 years ago

0328f858

Merge pull request #4441 from jakevdp:choice

a-googler committed 5 years ago

5cfd0e9a

Improve errors for failed compilations w/ core.concrete_or_error

jakevdp committed 5 years ago

492889f4

Fix rendering problems in lax.cond docstring.

hawkinsp committed 5 years ago

57324283

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

Older