ROCm/jax

Pull Requests Commits

fix an issue with newer versions of pytype

mattjj committed 5 years ago

12bde1b9

remove the lower_fun default multiple_results=True (#3524)

mattjj committed 5 years ago

Verified 2f7108f7

Add instructions for how to use the TensorBoard profiler to the profiling docs. (#3481)

skye committed 5 years ago

Verified 8fe6da08

Add private class wrapper for double-double arithmetic (#3521)

jakevdp committed 5 years ago

Verified 71023227

Fix typo understanding jaxprs page on readthedocs (#3513)

IgorWilbert committed 5 years ago

Verified e5d4ca31

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

skye committed 5 years ago

Verified 18798a30

fix jet typo: jnp not np (#3507)

mattjj committed 5 years ago

Verified a81e732d

Update JAX FAQ.rst, jax.device_put (#3496)

8bitmp3 committed 5 years ago

Verified f8570ec0

pin numpy version in setup.py to avoid warnings (#3509)

mattjj committed 5 years ago

Verified 3fff8379

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

Older