ROCm/jax

Pull Requests Commits

update jaxlib version in readme

mattjj committed 6 years ago

c760b05f

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

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

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

reset jax_numpy_rank_promotion to "allow" default

mattjj committed 6 years ago

Verified 1cd37bd9

Merge pull request #1253 from google/custom-transforms-vjp-nones

mattjj committed 6 years ago

Verified 449e27b3

leave todos for better error messages

mattjj committed 6 years ago

dbe56c30

fix for custom-transforms vjp nones bug

mattjj committed 6 years ago

6a81d81d

fix notebook link (fixes #1252)

mattjj committed 6 years ago

94b1dda4

Add jax.devices() and friends, and add `devices` arg to pmap.

skye committed 6 years ago

ae835b74

Merge pull request #1211 from levskaya/multibackend

skye committed 6 years ago

Verified 2d26ac38

Merge pull request #1250 from fehiepsi/dtype1

mattjj committed 6 years ago

Verified 89403229

cast float64 to canonical dtype in np.cov

fehiepsi committed 6 years ago

28df8a66

Older