ROCm/jax

Pull Requests Commits

ignore warnings in jax2tf tests

mattjj committed 5 years ago

e2ba29e3

Another small fix of the link rendering in the Autodiff Cookbook - vmap transformation (#3502)

8bitmp3 committed 5 years ago

Verified 7f9fc27c

Fix a link rendering to Autograd's reverse-mode Jacobian method (#3501)

8bitmp3 committed 5 years ago

Verified f0dff9c1

support stacked doubling (#3490)

jakevdp committed 5 years ago

Verified 1ed60b3f

implement jax.random.choice (#3463)

jakevdp committed 5 years ago

Verified 19f308b9

Allow specifying both `devices` and `axis_size` to pmap. (#3475)

skye committed 5 years ago

Verified 8f4ba7e6

Bump jaxlib version to 0.1.49 and update WORKSPACE (#3495)

skye committed 5 years ago

Verified a088c023

Fix test failure in jax2tf due to conflicting merges. (#3492)

hawkinsp committed 5 years ago

Verified 925d6611

Add random_gamma_grad and use in jax.random.gamma (#3281)

srvasude committed 5 years ago

Verified 927c2091

Improve dtype test coverage for random_test (#3254)

jakevdp committed 5 years ago

Verified 0da7b4d1

Disable the workaround to prevent expansion of type aliases (#3485)

gnecula committed 5 years ago

Verified 57d5a390

[jax2tf] Control on which TF device the TF converted code runs. (#3482)

gnecula committed 5 years ago

Verified 3b4a1230

Avoid tuple psum in pmean (#3479)

jekbradbury committed 5 years ago

Verified a05263f5

Remove documentation presubmit action now that the RTD presubmit is enabled. (#3480)

skye committed 5 years ago

Verified 68dcbdd1

Use myst to parse markdown docs, and convert a page from rst to markdown. (#3477)

skye committed 5 years ago

Verified 8bc78205

Attach source info to Jaxpr equations. (#3421)

hawkinsp committed 5 years ago

Verified 3290e16a

disable new jax2tf TPU test (#3476)

jakevdp committed 5 years ago

Verified 3c6c0af4

Add experimental precision doubling transform (#3465)

jakevdp committed 5 years ago

Verified 5a74ebf2

use original numpy for shape calculations (#3474)

mattjj committed 5 years ago

Verified 5344ec53

Add polyder numpy function (#3403)

ashridh committed 5 years ago

Verified 5ee936fd

[jax2tf] Added special case for tf.pad. (#3462)

gnecula committed 5 years ago

Verified 4f21b935

Fixed bug with silent TF using 64-bit operations in 32-bit mode. (#3471)

gnecula committed 5 years ago

Verified ce782e61

add jet primitives, refactor tests (#3468)

jacobjinkelly committed 5 years ago

Verified 575216e0

update tpu readme not to pin jax / jaxlib versions (#3467)

mattjj committed 5 years ago

Verified e9ce700b

fix a bug from #3459 (#3466)

mattjj committed 5 years ago

Verified 20f4ec64

update jaxlib version in readme

froystig committed 5 years ago

df3b507c

jaxpr pretty-print: wrap equation RHS when the LHS is long

froystig committed 5 years ago

a70ba920

Merge pull request #3459 from google/simplify-remat-partial-eval

mattjj committed 5 years ago

Verified 711c93db

mattjj committed 5 years ago

dfdf05fe

added reviewer suggestion

mattjj committed 5 years ago

005958e1

Older