ROCm/jax

Pull Requests Commits

clean up symbolic zero handling in transpose rules

mattjj committed 5 years ago

8fef9e8a

Copybara import of the project:

a-googler committed 5 years ago

dd9526bd

Merge pull request #5030 from google:zero-fixes

a-googler committed 5 years ago

730e4f7e

clean up symbolic zero handling in transpose rules

mattjj committed 5 years ago

cd81117b

Merge pull request #5014 from apaszke:xmap-nested

a-googler committed 5 years ago

d328816a

Merge pull request #5019 from jakevdp:dtype-warning

a-googler committed 5 years ago

693b1217

X32: don't warn when builtin type passed as dtype

jakevdp committed 5 years ago

9e1ebb51

Merge pull request #5011 from google:unify-reduction-collectives

a-googler committed 5 years ago

9c3a61cd

silence weird type error

mattjj committed 5 years ago

8d884e24

mattjj committed 5 years ago

0965bc40

Merge pull request #5016 from google:debug-nans-stores

a-googler committed 5 years ago

94974e5b

Merge pull request #4835 from google:xmap

a-googler committed 5 years ago

f9f97f7f

when resetting stores in nan debugging, check None

mattjj committed 5 years ago

7266fcbd

Merge pull request #5008 from jakevdp:equal-nan

a-googler committed 5 years ago

5ac16d14

add link to XLA bug about complex dtype allreduce

mattjj committed 5 years ago

3053f4bc

cleanup: unify pmin/pmax implementations with psum

mattjj committed 5 years ago

80973185

address reviewer comments

mattjj committed 5 years ago

ebd51e12

fix doc test failure

mattjj committed 5 years ago

50cb604f

simplify vmap collectives from two sets of rules to one

mattjj committed 5 years ago

8057cf91

Merge pull request #5003 from cloudhan:strict_action_env

a-googler committed 5 years ago

9dbdba29

Add support for nested xmaps

apaszke committed 5 years ago

ef1045bd

Support equal_nan parameter in jnp.allclose

jakevdp committed 5 years ago

82aed620

Merge pull request #5005 from apaszke:xmap-primitive

a-googler committed 5 years ago

c7057d5f

Forbid pmap/soft_pmap/sharded_jit inside xmap

apaszke committed 5 years ago

5ee2de16

Merge pull request #5012 from jakevdp:asarray

a-googler committed 5 years ago

afd57e31

Call asarray() rather than array() to avoid host round-trips.

jakevdp committed 5 years ago

9d2f6148

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

Older