ROCm/jax

Pull Requests Commits

Disable eigh batching test until Jaxlib 0.1.15 is released.

hawkinsp committed 6 years ago

8e16b7cd

make jtu.skip_on_devices read jax_platform_name

mattjj committed 6 years ago

07bf5096

generalize pmap collective-permute tests

mattjj committed 6 years ago

c1f01222

Implement eigh batching.

hawkinsp committed 6 years ago

76a26de4

refine scan test granularity

mattjj committed 6 years ago

840c6607

fix join_pvals bug from cond

mattjj committed 6 years ago

48811fbb

Merge pull request #697 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 620f63df

add lattice instantiate-at logic, scan works!

mattjj committed 6 years ago

a300319f

Merge pull request #698 from hawkinsp/build

hawkinsp committed 6 years ago

Verified 946f8674

Fix build with Bazel 0.25.

hawkinsp committed 6 years ago

7f9e1809

Use DeviceOrdinals method rather than accessing Computation._device_ordinals directly.

hawkinsp committed 6 years ago

60918077

flip lattice conventions to confuse our enemies

mattjj committed 6 years ago

643c26fd

comment out failing tests (for a moment)

mattjj committed 6 years ago

71e9c36e

must guarantee progress on lattice...

mattjj committed 6 years ago

360e3975

add some PartialVal invariants

mattjj committed 6 years ago

085f06e4

bump version for pypi

mattjj committed 6 years ago

c4a0c3da

Merge pull request #692 from google/prng-slow

mattjj committed 6 years ago

Verified ae225b2d

make static_argnums cache on value when possible

mattjj committed 6 years ago

5aea10c7

add collectives: allreduce min/max, permute

mattjj committed 6 years ago

37f9c268

improve jax.random shape error messages

mattjj committed 6 years ago

ea8e414a

Merge pull request #687 from kroq-gar78/average

mattjj committed 6 years ago

Verified 91d9a064

Fix typo in lapack.pyx.

hawkinsp committed 6 years ago

acaf76c6

Fix typo in lapack.pyx.

hawkinsp committed 6 years ago

582d6541

Add segment_sum to the docs and fix its rendering.

hawkinsp committed 6 years ago

e6082d20

Merge pull request #688 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 9dcf7daa

remove const_env from check_jaxpr, add scan trans

mattjj committed 6 years ago

c08b9fee

Changes for compatibility with a upcoming Jaxlib update.

hawkinsp committed 6 years ago

367833be

must handle literals + closure conversion better..

mattjj committed 6 years ago

25ac1987

tweak scan docstring

mattjj committed 6 years ago

54585743

tweaks to scan docstring

mattjj committed 6 years ago

a8ebc249

Newer Older