ROCm/jax

Pull Requests Commits

improve implementation of MVN logpdf

mattjj committed 5 years ago

3194a734

make lax_linalg.solve_triangular allow vector rhs

mattjj committed 5 years ago

93d3e347

Trigger a Travis build (#2477)

gnecula committed 5 years ago

Verified fcdbe63f

Pytype2 (#2476)

gnecula committed 5 years ago

Verified e41a5ea6

Added type annotations and removed unused imports (#2472)

gnecula committed 5 years ago

Verified 428377af

Added pytype checking to Travis (#2475)

gnecula committed 5 years ago

Verified 9331fc5b

Fix return type for vjp. (#2462)

hawkinsp committed 5 years ago

Verified 578e5cf6

Fix nondeterminism in test ordering in Python 3.6. (#2460)

hawkinsp committed 5 years ago

Verified 592ba02c

fix test errors from previous commit

mattjj committed 5 years ago

7f8ce8ff

make jaxpr pretty-print show multiple outputs

mattjj committed 5 years ago

1d0b7e2b

Update jaxlib version in README.md. (#2461)

hawkinsp committed 5 years ago

Verified c3f89090

Expose jax.lax.all_gather (#2449)

trevorcai committed 5 years ago

Verified d11a9ab1

Update JAX to use XLA hyperbolic functions. (#2415)

srvasude committed 5 years ago

Verified c7f211d4

Add more return types to api.py. (#2452)

hawkinsp committed 5 years ago

Verified afdd1a73

Increment jaxlib version to 0.1.42. (#2457)

hawkinsp committed 5 years ago

Verified cecfb37e

Increased tolerance for testScipySpecialFun (#2454)

gnecula committed 5 years ago

Verified 78c1f6b0

Changed to pmap_benchmark to make it runnable in Google (#2448)

gnecula committed 5 years ago

Verified cd7ab0a9

Updated Common Gotchas (#2435)

gnecula committed 5 years ago

Verified 2998a215

Add mypy type checking (#2430)

hawkinsp committed 5 years ago

Verified 68b32bf7

Fix xlog1py and xlogy not returning 0 when x == 0. (#2450)

sharadmv committed 5 years ago

Verified cd248ba3

Drop support for Python 3.5. (#2445)

hawkinsp committed 5 years ago

Verified cbdf9a5a

Merge pull request #2447 from google/remove-safe-mul

mattjj committed 5 years ago

Verified 7f1e8594

Wrap pad_widths in a tuple to avoid cache misses (#2379)

jekbradbury committed 5 years ago

Verified 26d5a68f

remove safe_mul (undo #383, also cf. #1052)

mattjj committed 5 years ago

f1d9130f

Add pmap_benchmark.py (#2409)

skye committed 5 years ago

Verified 75077a14

Update jax version number. (#2444)

hawkinsp committed 5 years ago

Verified 75813f46

Update changelog for jax 0.1.61 release. (#2443)

hawkinsp committed 5 years ago

Verified db8bea4c

Ensure jax.host_ids() returns a stable ordering (#2442)

trevorcai committed 5 years ago

Verified 84140ea5

Remove runtime tuple use from JAX. (#2441)

hawkinsp committed 5 years ago

Verified e46a002e

Fix Python 3.5 support. (#2439)

hawkinsp committed 5 years ago

Verified 985d5f73

Older