ROCm/jax

Pull Requests Commits

[ROCm]: Add support to continue on fail

Rahul Batra committed 1 year ago

dcaab558

add cholesky changes in bazel

a-googler committed 1 year ago

25ca6d84

add cholesky changes in bazel

Ruturaj4 committed 1 year ago

3aa80523

Update XLA dependency to use revision

a-googler committed 1 year ago

b8cec530

Merge pull request #21222 from tempoxylophone:patch-1

a-googler committed 1 year ago

9af0a317

Merge pull request #21221 from nouiz:doc

a-googler committed 1 year ago

48b90181

Merge pull request #21226 from Micky774:linalg-depr

a-googler committed 1 year ago

1dff5b37

Merge pull request #21227 from jakevdp:eye-offset

a-googler committed 1 year ago

d4f36e25

Cholesky rank-1 update kernel for JAX.

a-googler committed 1 year ago

e8b06ccf

Added a very rough sketch of Mosaic GPU lowering for Pallas

superbobry committed 1 year ago

e2918ca1

jnp.eye: allow k to be dynamic

jakevdp committed 1 year ago

3024c782

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

tlongeri committed 1 year ago

0ad5167d

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

Micky774 committed 1 year ago

5cc255b7

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

a-googler committed 1 year ago

0501d3d7

Finalize deprecations of several APIs

Jake VanderPlas committed 1 year ago

bb5787da

Reverts 85e91c2be4310d9728f7bfeefef921ee4a075135

ashishenoyp committed 1 year ago

1d6ffded

kl_div: fix incorrect formula in doc

jakevdp committed 1 year ago

c3c23932

Remove now-empty jax/_src/third_party/numpy

Jake VanderPlas committed 1 year ago

1e3a4b53

Fix type in advanced-autodiff.md

tempoxylophone committed 1 year ago

Verified 4bcad151

Update doc with the other error that can be thrown.

nouiz committed 1 year ago

f578d78f

Merge pull request #21216 from jakevdp:pmap-prng

a-googler committed 1 year ago

aea6b320

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

a-googler committed 1 year ago

95c4ba96

Fix sparse dot metadata loader

sergeykozub committed 1 year ago

84774b39

Fix PRNGKey handling under jit-of-pmap

jakevdp committed 1 year ago

5150cfee

Merge pull request #21215 from jakevdp:setops-docs

a-googler committed 1 year ago

e735a00c

Update XLA dependency to use revision

a-googler committed 1 year ago

98af7863

Merge pull request #21211 from elfiegg:main

a-googler committed 1 year ago

a56eb568

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

a-googler committed 1 year ago

6157d8e7

Improve docs for jax.numpy set-like operations

jakevdp committed 1 year ago

7ed7780b

Enable JAX memory tests for GPUs and CPUs

junwhanahn committed 1 year ago

cd6e0123

Older