ROCm/jax

Pull Requests Commits

mattjj committed 5 years ago

534c2c84

source line info experiments

mattjj committed 5 years ago

fe553f3c

omnistaging wip

mattjj committed 5 years ago

b7452a1c

Moved shift_right implementation from tfxla to jax_to_tf (#3378)

gnecula committed 5 years ago

Verified b3c348cc

Merge pull request #3395 from google/tweak-tolerance

mattjj committed 5 years ago

Verified e6a08e29

tweak t logpdf tolerance for float64

mattjj committed 5 years ago

650fb494

Merge pull request #3390 from google/issue3389

mattjj committed 5 years ago

Verified b7e945fc

fix defjvps with None as first rule

mattjj committed 5 years ago

0011fd5e

jaxpr: support dropped assignment

froystig committed 5 years ago

15bc6220

Disable some masking tests on TPU. (#3387)

hawkinsp committed 5 years ago

Verified 43c50833

Revert "Initial implementation of variadic lax.reduce() (#3342)" (#3384)

jakevdp committed 5 years ago

Verified 71461a37

Fix failures in masking_test.py (#3386)

hawkinsp committed 5 years ago

Verified fe1e1041

Update jax version to 0.1.70 (#3383)

skye committed 5 years ago

Verified 3e4210c3

Initial implementation of variadic lax.reduce() (#3342)

jakevdp committed 5 years ago

Verified 99401c5a

fix typo in docs

mattjj committed 5 years ago

d3ccf0a5

Merge pull request #3331 from JuliusKunze/mask-split

mattjj committed 5 years ago

Verified 307701c7

Merge pull request #3322 from j-towns/mask-of-jit

mattjj committed 5 years ago

Verified b7175a3f

Fix imports for jax_to_tf tests (#3377)

gnecula committed 5 years ago

Verified 67927b07

Refactor the jax_to_tf tests to separate the primitive test harness (#3376)

gnecula committed 5 years ago

Verified c12541d5

Fix remaining flakes and use exclude within setup.cfg (#3371)

jakevdp committed 5 years ago

Verified 6aa8f246

minor fixes from #3222 (#3368)

mattjj committed 5 years ago

Verified 563b65e8

mattjj committed 5 years ago

4ed8cce9

yet another doc fix

mattjj committed 5 years ago

ee428008

fix another docs bug

mattjj committed 5 years ago

2a6d3f41

improve broadcast handling in batching.py

mattjj committed 5 years ago

04191ab9

avoid some trival operations in mask computations (#3364)

mattjj committed 5 years ago

Verified 3b3e7a29

mattjj committed 5 years ago

0a716978

fix a couple ad_util.Zero type checks

mattjj committed 5 years ago

866c17c3

symbolic zeros tweak to work with div rule

mattjj committed 5 years ago

fb8b3a4d

clean up handling of aux data in jvp_subtrace_aux

mattjj committed 5 years ago

982f8670

Older