ROCm/jax

Pull Requests Commits

improve documetnation of lax parallel operators

mattjj committed 6 years ago

9b1af47a

Merge pull request #727 from google/pmap-improvements

mattjj committed 6 years ago

Verified e3d4213e

fix test condition

mattjj committed 6 years ago

056df10f

revise nested pmap gather to CRS-based impl

mattjj committed 6 years ago

f864edf7

make pmap catch device-count errors (+ tests)

mattjj committed 6 years ago

dad91933

Merge pull request #726 from google/dont-pack-all-outputs

mattjj committed 6 years ago

Verified 12036017

bump min jaxlib vers to 0.1.14 for tests to pass

mattjj committed 6 years ago

1116e984

avoid packing leaf outputs for jit/pmap funs

mattjj committed 6 years ago

b6031ffd

bump version for pypi

mattjj committed 6 years ago

751ca102

update minimum jaxlib version check to 0.1.13

mattjj committed 6 years ago

84aaf29c

bump version for pypi

mattjj committed 6 years ago

ab23bf6e

Initial FFT support.

skye committed 6 years ago

5d1c0145

Merge pull request #721 from google/update-jaxlib

mattjj committed 6 years ago

Verified c54ba844

fix randint docstring

mattjj committed 6 years ago

b68ed278

update jaxlib references to 0.1.15

mattjj committed 6 years ago

fc226dab

update jaxlib version number

mattjj committed 6 years ago

fb122540

mattjj committed 6 years ago

96ea32d4

Merge pull request #710 from rohan-anil/master

mattjj committed 6 years ago

Verified 60ef9293

mattjj committed 6 years ago

3b7a617e

fix typo in build_jaxlib_wheels_macos.sh

mattjj committed 6 years ago

c779e495

Merge pull request #718 from google/pmap-docs

mattjj committed 6 years ago

Verified 0e6aa3c4

typo fixes in pmap docstring

mattjj committed 6 years ago

52a2fb32

add pmap to sphinx docs

mattjj committed 6 years ago

d1994716

add pmap docstring

mattjj committed 6 years ago

38c6a8e8

Merge pull request #717 from hawkinsp/docs

hawkinsp committed 6 years ago

Verified be3358c2

Add some missing special functions to the documentation.

hawkinsp committed 6 years ago

8f461d26

Reenable some indexing tests that now pass.

hawkinsp committed 6 years ago

d9dc23a2

Update XLA build.

hawkinsp committed 6 years ago

bdb54e28

Merge pull request #713 from hawkinsp/master

hawkinsp committed 6 years ago

Verified eef97b6f

Expose jax.tree_util in the JAX docs.

hawkinsp committed 6 years ago

8c2c1a2e

Newer Older