ROCm/jax

Pull Requests Commits

fix DeviceTuple constant handler (fixes #831)

mattjj committed 6 years ago

4ab89084

improve error for unimplemented parallel primitive

mattjj committed 6 years ago

c061fd7b

Merge pull request #828 from google/pswapaxes-lowering

mattjj committed 6 years ago

Verified 424002bc

temporarily disable a check in pswapaxes

mattjj committed 6 years ago

d7c7df1b

add pswapaxes lowering (fixes #827)

mattjj committed 6 years ago

35de7047

fix broken travis path from previous commit

mattjj committed 6 years ago

d622e78e

pip wheels need to include the versionso use version '0.1.18' instead of 'latest'.

skye committed 6 years ago

04b0a911

Merge pull request #822 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 6e9c662a

Add bug number for integer dots.

hawkinsp committed 6 years ago

bd389b7f

Add support for integer dot operations.

hawkinsp committed 6 years ago

15361c4d

Merge pull request #821 from georgedahl/optimizer_state_saving

mattjj committed 6 years ago

Verified 8309836d

Changes to jax optimizers.py to facilitate serialization of optimizer state.

georgedahl committed 6 years ago

305127d8

Merge pull request #820 from google/simplify-gather-shape-rule

mattjj committed 6 years ago

Verified 1feefd10

Merge pull request #819 from jekbradbury/dot_batch_rule

mattjj committed 6 years ago

Verified 6cc87e0d

simplify gather shape rule

mattjj committed 6 years ago

9e49500c

fix bug in dot batching rule

jekbradbury committed 6 years ago

9bc5f2ae

Update jaxlib version

skye committed 6 years ago

a61d5f6e

Merge pull request #816 from google/vmap-of-pmap

mattjj committed 6 years ago

Verified 051abc1b

add process_map to vmap tracer, i.e. vmap-of-pmap

mattjj committed 6 years ago

e1f32d0d

Update jaxlib references to 'latest'

skye committed 6 years ago

3a234641

add future to macos wheel build script

mattjj committed 6 years ago

a12ca881

Update Dockerfile to pip install future for py2 build.

skye committed 6 years ago

3819eac2

Merge pull request #813 from hawkinsp/master

hawkinsp committed 6 years ago

Verified dd3b681b

Add documentation on asynchronous dispatch.

hawkinsp committed 6 years ago

91e6ba32

Update XLA workspace

skye committed 6 years ago

e071d3fd

skye committed 6 years ago

451e2081

bump version for pypi

mattjj committed 6 years ago

f3501eac

Merge pull request #811 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 0f1d62df

hawkinsp committed 6 years ago

22ad01ed

Merge pull request #809 from hawkinsp/block

hawkinsp committed 6 years ago

Verified 526c0298

Older