ROCm/jax

Pull Requests Commits

Fix dtype in scoring matching example

shoyer committed 5 years ago

020c4090

shoyer committed 5 years ago

bc48850e

fix tests for random dtype

shoyer committed 5 years ago

815f0615

Merge branch 'lax-shape-int' into remove-x64-mode

shoyer committed 5 years ago

acf9ceb1

Ignore NumPy scalars when determining dtypes

shoyer committed 5 years ago

07d07f16

revert remove x64 entirely; just change default

shoyer committed 5 years ago

e0a758e1

Merge branch 'lax-shape-int' into remove-x64-mode

shoyer committed 5 years ago

7c2c21a5

Allow Poly in shapes

shoyer committed 5 years ago

0c0b9076

Ensure ShapedArray.shape is always a tuple of builtins integers

shoyer committed 5 years ago

f41f4a87

fixup casting in arange

shoyer committed 5 years ago

486e50c9

WIP: an experiment to see if we can remove jax_enable_x64

shoyer committed 5 years ago

989ac623

Issue1635 expm (#1940)

sriharikrishna committed 5 years ago

03b2ae6d

Bump jaxlib version to 0.1.38 and update WORKSPACE.

skye committed 5 years ago

f04348ed

Fix Sysml paper link in README (#2036)

Mu Li committed 5 years ago

2c80cd3d

add test for issue #553

mattjj committed 5 years ago

17b5fe11

Minimal changes to make Jax pass a pytype check. (#2024)

hawkinsp committed 5 years ago

Verified 7dbc8dc1

Add jax changelog (#2022)

skye committed 6 years ago

Verified 19fb494a

Fix README typo (#2020)

j-towns committed 6 years ago

371001aa

[docs] Pmap compiles functions with XLA (#2021)

j-towns committed 6 years ago

3974df0a

changes loop_mjp(f, x, M) (#2013)

suryabulusu committed 6 years ago

71323b5d

Merge pull request #1980 from google/jvp-while

froystig committed 6 years ago

Verified 28f70cc8

test JVP of while loop, and fix the nonzero tangent calculation in the JVP rule

froystig committed 6 years ago

335ecb97

Implement shapecheck for more primitives (#1990)

juliuskunze committed 6 years ago

55c971e4

Add betainc to JAX (#1998)

srvasude committed 6 years ago

80b35dd4

[pmap] Add support for nested pmaps on multihost platforms via axis_size (#2002)

trevorcai committed 6 years ago

12975bbc

Defer to unrecognized types in arithmetic (#1942)

shoyer committed 6 years ago

Verified a5644edb

Update references to bazel repositories in WORKSPACE to match TF head. (#2005)

hawkinsp committed 6 years ago

Verified 653001aa

Use a uniform rng rather than a normal rng to defeat CSE. (#2000)

hawkinsp committed 6 years ago

Verified 11224bd2

Remove :libjax alias from BUILD file. (#1996)

hawkinsp committed 6 years ago

Verified 938a7f80

added batching to LAPACK triangular_solve (#1985)

akbir committed 6 years ago

dcda87d0

Older