ROCm/jax

Pull Requests Commits

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

fix flakes at head (#3361)

jakevdp committed 5 years ago

Verified 528207bd

Fix polyadd and test (#3344)

ashridh committed 5 years ago

Verified b1adef40

A couple of ad_util.zero were missed in #3222 (#3363)

gnecula committed 5 years ago

Verified 65d95f10

fix a one-character issue from a bad merge (#3362)

mattjj committed 5 years ago

Verified 508821cc

Update optix.py (#3355)

mtthss committed 5 years ago

Verified dc41ff90

Older