ROCm/jax

Pull Requests Commits

minor change as source sync test

froystig committed 5 years ago

5f6fab8d

Internal change

a-googler committed 5 years ago

6eb4f54e

Revert "[jax2tf] Added Getting Started notebook (#4460)" (#4463)

gnecula committed 5 years ago

Verified 93f4dd44

[jax2tf] Added Getting Started notebook (#4460)

gnecula committed 5 years ago

Verified 1d739527

Merge pull request #4457 from google:tycheck-invar-eqn-context

a-googler committed 5 years ago

2166f0ee

Merge pull request #4458 from hawkinsp:xla

a-googler committed 5 years ago

7efa3209

hawkinsp committed 5 years ago

49d851d2

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

Older