ROCm/jax

Pull Requests Commits

Merge pull request #834 from google/issue830

mattjj committed 6 years ago

Verified 97e16655

Merge pull request #836 from aslanides/master

mattjj committed 6 years ago

Verified 902f32d7

Fix reference to init_fun in optimizers docs.

aslanides committed 6 years ago

1a4f6aa5

np.arange shouldn't pop its kwargs (fixes #830)

mattjj committed 6 years ago

1829508b

Merge pull request #832 from google/issue831

mattjj committed 6 years ago

Verified 53241188

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

docstring improvements from @skye comments

mattjj committed 6 years ago

121d7812

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

mattjj committed 6 years ago

12b90ddf

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

mattjj committed 6 years ago

Verified 1feefd10

finish drafting defvjp/defjvp docstrings

mattjj committed 6 years ago

a6c41a32

improve custom_gradient docstring

mattjj committed 6 years ago

cfaa49f8

add docstrings for defjvp and defjvp2

mattjj committed 6 years ago

948ec8fb

add docstring for defjvp_all

mattjj committed 6 years ago

ab20f029

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

add custom_gradient

mattjj committed 6 years ago

720dec40

add docstring to custom_transforms

mattjj committed 6 years ago

372d60bb

make custom_transforms handle pytrees, add api.defvjp

mattjj committed 6 years ago

35e5e644

Newer Older