ROCm/jax

Pull Requests Commits

[ROCm]: Add support to continue on fail

Rahul Batra committed 2 years ago

dcaab558

add cholesky changes in bazel

a-googler committed 2 years ago

25ca6d84

add cholesky changes in bazel

Ruturaj4 committed 2 years ago

3aa80523

Update XLA dependency to use revision

a-googler committed 2 years ago

b8cec530

Merge pull request #21222 from tempoxylophone:patch-1

a-googler committed 2 years ago

9af0a317

Merge pull request #21221 from nouiz:doc

a-googler committed 2 years ago

48b90181

Merge pull request #21226 from Micky774:linalg-depr

a-googler committed 2 years ago

1dff5b37

Merge pull request #21227 from jakevdp:eye-offset

a-googler committed 2 years ago

d4f36e25

Cholesky rank-1 update kernel for JAX.

a-googler committed 2 years ago

e8b06ccf

Added a very rough sketch of Mosaic GPU lowering for Pallas

superbobry committed 2 years ago

e2918ca1

jnp.eye: allow k to be dynamic

jakevdp committed 2 years ago

3024c782

Add support for i1 vmasks with packed tiling and 16-bit comparisons (requires hardware support)

tlongeri committed 2 years ago

0ad5167d

Rename rcond/tol to rtol in linalg.matrix_rank and linalg.pinv

Micky774 committed 2 years ago

5cc255b7

Merge pull request #21223 from jakevdp:kl-div-doc

a-googler committed 2 years ago

0501d3d7

Finalize deprecations of several APIs

Jake VanderPlas committed 2 years ago

bb5787da

Reverts 85e91c2be4310d9728f7bfeefef921ee4a075135

ashishenoyp committed 2 years ago

1d6ffded

kl_div: fix incorrect formula in doc

jakevdp committed 2 years ago

c3c23932

Remove now-empty jax/_src/third_party/numpy

Jake VanderPlas committed 2 years ago

1e3a4b53

Fix type in advanced-autodiff.md

tempoxylophone committed 2 years ago

Verified 4bcad151

Update doc with the other error that can be thrown.

nouiz committed 2 years ago

f578d78f

Merge pull request #21216 from jakevdp:pmap-prng

a-googler committed 2 years ago

aea6b320

[JAX] Use first process id instead of process 0 to share multi-host data.

a-googler committed 2 years ago

95c4ba96

Fix sparse dot metadata loader

sergeykozub committed 2 years ago

84774b39

Fix PRNGKey handling under jit-of-pmap

jakevdp committed 2 years ago

5150cfee

Merge pull request #21215 from jakevdp:setops-docs

a-googler committed 2 years ago

e735a00c

Update XLA dependency to use revision

a-googler committed 2 years ago

98af7863

Merge pull request #21211 from elfiegg:main

a-googler committed 2 years ago

a56eb568

Merge pull request #21176 from jakevdp:shard-map-api-docs

a-googler committed 2 years ago

6157d8e7

Improve docs for jax.numpy set-like operations

jakevdp committed 2 years ago

7ed7780b

Enable JAX memory tests for GPUs and CPUs

junwhanahn committed 2 years ago

cd6e0123

Older