ROCm/jax

Pull Requests Commits

make static_argnums cache on value when possible

mattjj committed 6 years ago

5aea10c7

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

Changes for compatibility with a upcoming Jaxlib update.

hawkinsp committed 6 years ago

367833be

Fixes, and skip test if ZeroDivisionError

kroq-gar78 committed 6 years ago

4ef154a3

Average fixes. Doesn't work for non-empty shapes

kroq-gar78 committed 6 years ago

3468e87f

Started work on np.average

kroq-gar78 committed 6 years ago

725a4695

bump version for pypi, leave DeviceTuples off

mattjj committed 6 years ago

9788a358

Merge pull request #684 from google/device-tuples

mattjj committed 6 years ago

Verified 0ccf7cf6

Merge pull request #685 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 84327c13

Fix test case for nan_to_num when enable_x64 is False.

hawkinsp committed 6 years ago

fd53e394

Fix type mismatch for nan_to_num for 64-bit types. Fixes #683.

hawkinsp committed 6 years ago

6d77fb7d

use DeviceTuples in optimizer states again

mattjj committed 6 years ago

9fc47e51

fix tuple unpacking problems

mattjj committed 6 years ago

19e0f8de

avoid destructuring on len(sharded_device_tuple)

mattjj committed 6 years ago

535773fe

Merge pull request #682 from google/optimizer-utilities

mattjj committed 6 years ago

Verified e1b7cbc7

improve pmap error messages

mattjj committed 6 years ago

69030135

add optimizer utilities (fixes #244 and #681)

mattjj committed 6 years ago

e7511893

Merge pull request #671 from google/index-ops-advanced-indexing

mattjj committed 6 years ago

Verified b226cbc5

reviewer-suggested fixes

mattjj committed 6 years ago

3ab52646

fix up the is_advanced_int_indexer logic

mattjj committed 6 years ago

41a7a944

add advanced indexing support to jax index ops

mattjj committed 6 years ago

9adfb806

skip a test on tpu

mattjj committed 6 years ago

b5d95f8b

Merge pull request #680 from google/disable-devicetuples

mattjj committed 6 years ago

Verified eb4ac933

update pmap to flatten correctly (was a perf bug)

mattjj committed 6 years ago

bf6c15b5

Update XLA to pick up nccl performance improvement.

skye committed 6 years ago

5033fe86

Merge pull request #675 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 57ad3746

Older