ROCm/jax

Pull Requests Commits

Avoid tuple psum in pmean

jekbradbury committed 5 years ago

Verified 8d92c9b7

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

Merge pull request #3396 from JuliusKunze/simplify-scan-mask

mattjj committed 5 years ago

Verified 140c9ea9

Merge pull request #3456 from jacobjinkelly/int_pow

mattjj committed 5 years ago

Verified 270e9214

Merge pull request #3455 from jacobjinkelly/erf_inv

mattjj committed 5 years ago

Verified 60760f27

Merge pull request #3429 from BuddenD/changelist/316251368

mattjj committed 5 years ago

Verified c30d23f4

Merge branch 'master' into changelist/316251368

mattjj committed 5 years ago

Verified fe14aa3e

Merge pull request #3426 from google/warn-jit-of-pmap

mattjj committed 5 years ago

Verified 08b83e84

print warning when doing jit-of-pmap

mattjj committed 5 years ago

d4c6cb62

mattjj committed 5 years ago

8a901ba0

add scipy.special.zeta, scipy.special.polygamma (#3385)

fehiepsi committed 5 years ago

Verified 5d20c6fd

simplify remat partial eval parameterization

mattjj committed 5 years ago

05e0716f

add int pow rule

jacobjinkelly committed 5 years ago

f463598f

Fix jnp.arange(x) for fractional x (#3453)

jakevdp committed 5 years ago

Verified 6bcf0568

add erf inv rule

jacobjinkelly committed 5 years ago

3cf6b1de

Merge pull request #3449 from google/issue3440

mattjj committed 5 years ago

Verified 37f4722f

Older