ROCm/jax

Pull Requests Commits

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

Make xmap into a primitive

apaszke committed 5 years ago

c5433b0e

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

use strict action env on windows to avoid constant full rebuilding

cloudhan committed 5 years ago

ea340eed

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

Fix integer overflow in np.split on windows.

hawkinsp committed 5 years ago

a07c5e11

Support 'initial' argument in jax.numpy reductions

jakevdp committed 5 years ago

841ea14c

Merge pull request #4970 from jakevdp:reduction-tests

a-googler committed 5 years ago

a96aca5e

Add support for multi-host partitioning when using pmap(sharded_jit).

skye committed 5 years ago

4e56cf96

Merge pull request #4988 from hawkinsp:windows

a-googler committed 5 years ago

7d135648

Newer Older