ROCm/jax

Pull Requests Commits

tweaks to save precious vertical space

mattjj committed 6 years ago

76a1bb0c

Merge pull request #1605 from google/memleaks

mattjj committed 6 years ago

Verified 69238600

Implement Poisson distribution for jax.scipy.stats (#1608)

TuanNguyen27 committed 6 years ago

a9b110f0

improve names, avoid double lookup (thanks @hawkinsp)

mattjj committed 6 years ago

d2156ea1

fix a leak where compiled results lived too long

mattjj committed 6 years ago

8bcee8d4

Merge pull request #1613 from google/vmap-improvements

mattjj committed 6 years ago

Verified 2bab332e

mattjj committed 6 years ago

46fe76c2

make vmap structured axes work for any pytree

mattjj committed 6 years ago

979b3835

add test case per reviewer comment

mattjj committed 6 years ago

d09571eb

check full error message

mattjj committed 6 years ago

213b899e

mattjj committed 6 years ago

8aec519c

Merge branch 'vmap-improvements' of github.com:google/jax into vmap-improvements

mattjj committed 6 years ago

9923cefe

address reviewer comments, fix test error

mattjj committed 6 years ago

14acca7b

Update jax/api.py

mattjj committed 6 years ago

Verified 9d94c423

improve vmap error messages

mattjj committed 6 years ago

eae47b23

Update jaxlib build scripts to build Python 3.8.0 wheels. (#1612)

hawkinsp committed 6 years ago

Verified b3e4a1a8

Update XLA to fix another CUDA build problem. (#1610)

hawkinsp committed 6 years ago

Verified 33e5d073

Update XLA to fix CUDA build failure. (#1609)

hawkinsp committed 6 years ago

Verified cba5f229

Update XLA. (#1607)

hawkinsp committed 6 years ago

Verified 5858983f

Add trivial implementations of eigvals/eigvalsh (#1604)

stephentu committed 6 years ago

39daf07d

allow unmapped vmap args to be arbitrary objects

mattjj committed 6 years ago

cbadfd41

improve vmap docstring and tree prefix errors

mattjj committed 6 years ago

f5079a62

Increment jax and jaxlib versions. (#1603)

hawkinsp committed 6 years ago

Verified affa2dcc

Merge pull request #1585 from fehiepsi/mxstep

mattjj committed 6 years ago

Verified a986463c

Disable LaxControlFlowTest.test_custom_root_with_custom_linear_solve. (#1601)

hawkinsp committed 6 years ago

Verified 57acdfe9

Use log1p in definition of logaddexp2 to match logaddexp. (#1599)

hawkinsp committed 6 years ago

Verified 97944a40

Add np.fft.ifftn (#1594)

shoyer committed 6 years ago

89c90923

Fix definition of jax.numpy.divmod for floating-point types. (#1597)

hawkinsp committed 6 years ago

Verified fcf5633f

Merge pull request #1434 from joaogui1/better-documentation

gnecula committed 6 years ago

Verified d7bdbdff

Add some type helpers to lax_numpy. (#1593)

hawkinsp committed 6 years ago

Verified f7a44523

Older