ROCm/jax

Pull Requests Commits

reviewer comments: no 'install' text needed

mattjj committed 6 years ago

a0095e34

update notebooks b/c jax comes with colab now!

mattjj committed 6 years ago

b672ad72

Merge pull request #1296 from trevorcai/poly_decay

mattjj committed 6 years ago

Verified be4dc0eb

Merge pull request #1272 from alexalemi/vectorize

mattjj committed 6 years ago

Verified c5737d0f

Alex Alemi committed 6 years ago

8b98f8e2

Merge pull request #1295 from google/more-lax-autodiff-tests

mattjj committed 6 years ago

Verified 110634d5

move asinh/acosh/atanh to lax_numpy.py only

mattjj committed 6 years ago

cac042c3

add special value grad tests, sinh failing

mattjj committed 6 years ago

42258571

Merge pull request #1277 from google/primitive-shape-error-messages

mattjj committed 6 years ago

Verified 1147eac3

if we hit an XLA Build error, it's a JAX bug

mattjj committed 6 years ago

f5c152b7

fix typo (thanks @dpfau)

mattjj committed 6 years ago

2b561c7c

bump jax version for pypi

mattjj committed 6 years ago

731e46b3

Merge pull request #1269 from google/issue1267

mattjj committed 6 years ago

Verified 5cb23157

remove jaxlib import (broke google3 pulldown)

mattjj committed 6 years ago

c0627d81

switch rolled/unrolled loops in prng hash

mattjj committed 6 years ago

2815c55b

avoid Calls inside While/Cond

mattjj committed 6 years ago

478832c9

Merge pull request #1268 from fehiepsi/condfix

mattjj committed 6 years ago

Verified 29b5d956

address comments

fehiepsi committed 6 years ago

b7edc39b

Changed axis kwarg in wrapper for py2 compat.

Alex Alemi committed 6 years ago

e6854009

added documentation

Alex Alemi committed 6 years ago

55a06cbd

fixing issues with tests

Alex Alemi committed 6 years ago

54f28ed5

Add polynomial_decay schedule to optimizers

trevorcai committed 6 years ago

ae52f865

use lax instead of numpy to avoid circular import

fehiepsi committed 6 years ago

16bd2725

fix cond batching rule issues

fehiepsi committed 6 years ago

92d85a89

Alex Alemi committed 6 years ago

4ceebcf4

Merge pull request #1265 from jlebar/quick-start-notebook2

mattjj committed 6 years ago

Verified c3e52cfc

Use np.eye instead of onp.eye in quickstart notebook.

Justin Lebar committed 6 years ago

fe683acd

add try/finally to pxla global state management

mattjj committed 6 years ago

fd9e0333

update jaxlib version for macOS wheels

mattjj committed 6 years ago

009ff074

Workaround to make XRT work with multi-backend change.

skye committed 6 years ago

Verified d5820953

Older