ROCm/jax

Pull Requests Commits

document pmap with pytrees, fixes #1486

mattjj committed 6 years ago

585cefc8

Merge pull request #1513 from google/travis-conda-build-issue

mattjj committed 6 years ago

Verified 1f45b5ca

try following jupyterlab/jupyter-renderers#212 to fix travis

mattjj committed 6 years ago

0eea9885

Delete experimental/lapax. (#1511)

hawkinsp committed 6 years ago

Verified 3711e0cf

Move lax.tie_in inside lax.full_like onto the fill value instead the output of lax.full. (#1507)

hawkinsp committed 6 years ago

Verified 06178d29

Describe minimum OS version requirements in README.md. (#1508)

hawkinsp committed 6 years ago

Verified 9ff9e9e3

Add text to documentation and to jaxlib version check suggesting a pip update. (#1506)

hawkinsp committed 6 years ago

Verified d161e1ba

Merge pull request #1505 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 5457775a

Add Token to core.pytype_aval_mappings.

hawkinsp committed 6 years ago

3035bf72

Use collections.abc.Sequence in favor of collections.Sequence (#1504)

shoyer committed 6 years ago

Verified d338449e

Merge pull request #1499 from joaogui1/readme

jekbradbury committed 6 years ago

Verified 4479e4b4

Fix List rendering in RTD for primitives.ipynb (#1501)

trevorcai committed 6 years ago

c50495b2

Remove unnecessary zero checks from JVP rules in lax_linalg (#1490)

shoyer committed 6 years ago

Verified 16219358

Merge pull request #1422 from j-towns/toposort-rm-dupes

hawkinsp committed 6 years ago

Verified 2c596a9d

Fixes Readme links to notebooks

joaogui1 committed 6 years ago

d6bc62bd

Add a separate build matrix entry for documentation testing. (#1495)

gnecula committed 6 years ago

Verified 858a4119

Merge pull request #1492 from joaogui1/dirichlet

jekbradbury committed 6 years ago

Verified 164cb4e2

Change device_put to take a `device` argument instead of `device_num`. (#1463)

skye committed 6 years ago

Verified 5585dda9

Adds dirichlet to __init__.py, making it usable

joaogui1 committed 6 years ago

245f4067

Fix rendering of jax.eval_shape docs.

hawkinsp committed 6 years ago

27afa128

Merge pull request #1489 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 4ca8576e

Update XLA release to include GPU memory corruption fix.

hawkinsp committed 6 years ago

d238b1f4

Document all_to_all and ppermute.

hawkinsp committed 6 years ago

78132c15

Merge pull request #1485 from hawkinsp/master

hawkinsp committed 6 years ago

Verified dd72fd5d

Emit a better error if mismatched axis sizes are passed to pmap.

hawkinsp committed 6 years ago

66098189

Speedup JVP for triangular solve (#1466)

shoyer committed 6 years ago

Verified c8247786

Merge pull request #1481 from hawkinsp/master

hawkinsp committed 6 years ago

Verified e9f99189

Register OrderedDict as a pytree type.

hawkinsp committed 6 years ago

fffec814

Merge pull request #1474 from gnecula/documentation

gnecula committed 6 years ago

Verified b2493a1e

Merge pull request #1478 from hawkinsp/infeed

hawkinsp committed 6 years ago

Verified 4a075be6

Older