ROCm/jax

Pull Requests Commits

Add support for pmap in_axes other than 0 and None

apaszke committed 5 years ago

a5bc7353

Internal change

a-googler committed 5 years ago

bdd79156

cleanup: remove unused test arg

jakevdp committed 5 years ago

898fa7e6

Merge pull request #4848 from apaszke:pmap-in-axes

a-googler committed 5 years ago

342d62dd

Add support for pmap in_axes other than 0 and None

apaszke committed 5 years ago

6914058c

Rm old attribute annotations from TraceStack

j-towns committed 5 years ago

b8920a11

Merge pull request #4851 from hawkinsp:nn

a-googler committed 5 years ago

08115e89

Merge pull request #4852 from sharadmv:patch-5

a-googler committed 5 years ago

34d2e11b

Fix typo in FAQ

sharadmv committed 5 years ago

Verified a51afa30

Merge pull request #4836 from jblespiau:changelist/341311408

a-googler committed 5 years ago

ef6d5c4d

Merge pull request #4830 from shoyer:slow-jit-faq

a-googler committed 5 years ago

09b2c059

Add a FAQ entry on slow JIT compilation

shoyer committed 5 years ago

bb684b59

Delete jax.nn.functions.

hawkinsp committed 5 years ago

c9f4b274

Merge pull request #4847 from hawkinsp:py39

a-googler committed 5 years ago

a3a55a91

Add Python 3.9 support to jaxlib build.

hawkinsp committed 5 years ago

2c6f932e

Merge pull request #4814 from apaszke:sharding-spec

a-googler committed 5 years ago

307b528b

Add some type annotations in parallel_callable

apaszke committed 5 years ago

f6372630

Make ShardingSpec more flexible

apaszke committed 5 years ago

8c1fbdc9

Merge pull request #4844 from hawkinsp:manylinux

a-googler committed 5 years ago

08131d4e

Fix manylinux2010 compliance of GPU wheels.

hawkinsp committed 5 years ago

dadf732f

DOC: fix generating code in type_promotion.rst

jakevdp committed 5 years ago

d913e6b3

Merge pull request #4846 from zhangqiaorjc:profiler_fix_note

a-googler committed 5 years ago

78e7095d

Add a CHANGELOG note about profiler bug fix.

zhangqiaorjc committed 5 years ago

5e850958

Merge pull request #4822 from ayush-1506:4775

a-googler committed 5 years ago

ca766caa

Merge pull request #4845 from hawkinsp:xla

a-googler committed 5 years ago

dcc81c33

hawkinsp committed 5 years ago

a4669370

Merge pull request #4841 from hawkinsp:optix

a-googler committed 5 years ago

9b3f1634

Merge pull request #4831 from mariosasko:fix-readme-link

a-googler committed 5 years ago

371efd3c

Merge pull request #4828 from j-towns:api-doc-fixes

a-googler committed 5 years ago

ffff3a42

Merge pull request #4840 from hawkinsp:adam

a-googler committed 5 years ago

4003ada3

Newer Older