ROCm/jax

Pull Requests Commits

Renamed experimental/jax_to_tf to experimental/jax2tf (#3404)

gnecula committed 5 years ago

Verified 27906ce2

Add np.triu_indices_from function (#3346)

ashridh committed 5 years ago

Verified 832431db

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

WIP: print rules

shoyer committed 5 years ago

52184699

Fix tree-splitting check

shoyer committed 5 years ago

e3d077ef

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

Newer Older