ROCm/jax

Pull Requests Commits

minor refactoring useful for pdot

mattjj committed 5 years ago

483d0717

fix doc test failure

mattjj committed 5 years ago

d2c28aeb

simplify vmap collectives from two sets of rules to one

mattjj committed 5 years ago

6523b884

Merge pull request #4989 from jakevdp:gradient-args

a-googler committed 5 years ago

30bb4cf5

Make jnp.gradient() call signature match np.gradient

jakevdp committed 5 years ago

bdaae722

Merge pull request #5007 from jakevdp:avoid-copy

a-googler committed 5 years ago

c9579b85

Merge pull request #4919 from google:update-readme-citation

a-googler committed 5 years ago

2d9a1072

Merge pull request #4904 from apaszke:pmap-out-axes

a-googler committed 5 years ago

43549157

Merge pull request #5004 from dpfau:complex_slogdet

a-googler committed 5 years ago

cb8d39f2

jnp.array(): avoid unnecessary numpy array copy

jakevdp committed 5 years ago

3845712a

Add support for non-zero (but still not-None) out_axes in pmap

apaszke committed 5 years ago

5879967c

Add complex types to gradient of slogdet

dpfau committed 5 years ago

8d1daba9

Merge pull request #4980 from apaszke:use-taggedtuple

a-googler committed 5 years ago

ad6ceeea

Use taggedtuple instead of namedtuple when defining ShardingSpecs

apaszke committed 5 years ago

94fcd7f2

Merge pull request #4954 from apaszke:xmap-xla

a-googler committed 5 years ago

c7a08b92

Add XLA lowering for xmap

apaszke committed 5 years ago

2494e0c3

Merge pull request #4999 from cloudhan:wheel-win-fix

a-googler committed 5 years ago

cc2ec614

use shutil.copyfile to avoid readonly pyd file

cloudhan committed 5 years ago

2146cea1

Merge pull request #4941 from j-towns:eigvals-jvp

a-googler committed 5 years ago

c7b2b9ed

Merge pull request #4990 from hawkinsp:windows2

a-googler committed 5 years ago

b651d7b9

Merge pull request #4982 from hawkinsp:wheel

a-googler committed 5 years ago

83e7e393

Merge pull request #4995 from hawkinsp:xla

a-googler committed 5 years ago

5fc47c8a

hawkinsp committed 5 years ago

ba84168c

Merge pull request #4993 from jakevdp:fix-random-doc

a-googler committed 5 years ago

1f536d33

Merge pull request #4991 from jakevdp:reduction-where

a-googler committed 5 years ago

82135b53

DOC: fix jax.random documentation

jakevdp committed 5 years ago

4ae393f8

Merge pull request #4984 from bchetioui:merge_prs

a-googler committed 5 years ago

7bd67efc

Merge pull request #4972 from skye:pod_sharded_jit_pmap

a-googler committed 5 years ago

e177acec

Support 'where' argument in jax.numpy reductions

jakevdp committed 5 years ago

cfc2ff4a

Merge pull request #4975 from jakevdp:reduction-initial

a-googler committed 5 years ago

4a3bc901

Older