ROCm/jax

Pull Requests Commits

Add transparent copy of buffers in the C++ jax.jit.

jblespiau committed 5 years ago

5583f652

Merge pull request #4468 from gnecula:tf_colab4

a-googler committed 5 years ago

0290339e

Another attempt to save the notebook

gnecula committed 5 years ago

ae7e5b99

Updated the notebook with a run on a public runtime

gnecula committed 5 years ago

27e5c2a7

gnecula committed 5 years ago

50ce9150

[jax2tf] Add the getting started notebook

gnecula committed 5 years ago

ed0a098d

Merge pull request #4467 from google:update-pypi

a-googler committed 5 years ago

2dfc5f96

update version and changelog for pypi

mattjj committed 5 years ago

1c5a09ac

Merge pull request #4464 from google:minor-source-sync

a-googler committed 5 years ago

cf1fefcd

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

Older