ROCm/jax

Pull Requests Commits

[ROCm]: Add missing hipStreamWaitEvent API call

Rahul Batra committed 2 years ago

85750555

Update XLA dependency to use revision

a-googler committed 2 years ago

604f5eca

Add option to set coordinator lookup timeout for TPU clusters

a-googler committed 2 years ago

c5869feb

Merge pull request #20320 from jakevdp:key-reuse-util

a-googler committed 2 years ago

33e1a962

Merge pull request #20323 from jakevdp:fix-doctest

a-googler committed 2 years ago

604cdbcf

[Mosaic] Restore Python pipeline and add a CLI flag to run it.

tlongeri committed 2 years ago

99fadcbc

jaxlib: Add `ifrt_proxy.pyi` to `build_wheel.py`.

a-googler committed 2 years ago

df9cefab

Merge pull request #20322 from jakevdp:complex-plane-numpy2

a-googler committed 2 years ago

6fa75aad

doctest: avoid modifying global flag state

jakevdp committed 2 years ago

4963ae16

lax_test: make ComplexPlane sign tests compatible with numpy 2

jakevdp committed 2 years ago

9062cfba

Fix link in the error message.

Jieying Luo committed 2 years ago

cc068360

Merge pull request #20307 from jakevdp:update-api-tests

a-googler committed 2 years ago

e6332d3a

[key reuse] add internal function_type_signature utility

jakevdp committed 2 years ago

332450ba

Merge pull request #20314 from jakevdp:dep-maps

a-googler committed 2 years ago

d95df141

Skip offloading tests when used together with an old jaxlib

apaszke committed 2 years ago

e5a16a01

More numerically stable `jax.experimental.jet._integer_pow_taylor`.

a-googler committed 2 years ago

3830b172

Remove internal references to deprecated jax.experimental.maps

jakevdp committed 2 years ago

84e49bd6

Register maps module deprecation outside of module

Jake VanderPlas committed 2 years ago

d8662886

Require array-like inputs to sparse_plus

Jake VanderPlas committed 2 years ago

b48aec57

Strip device_assignment on GPU platform.

a-googler committed 2 years ago

0b28a4b1

Update XLA dependency to use revision

a-googler committed 2 years ago

a4533010

[XLA:Python] Improve error checking for the return value of the to_iterable function of custom pytree nodes.

hawkinsp committed 2 years ago

f7594522

Remove outdated documentation section,

cheshire committed 2 years ago

0e1b3e5b

[array-api] update to latest array_api_tests commit

jakevdp committed 2 years ago

dad31cf8

Merge pull request #20258 from Gattocrucco:digitize-fix-corner-case

a-googler committed 2 years ago

cfb968d2

Merge pull request #20250 from jakevdp:key-reuse-jit

a-googler committed 2 years ago

d8f231a8

Merge pull request #19602 from chaserileyroberts:chase/pcollective_broadcast

a-googler committed 2 years ago

d6588ac5

chaserileyroberts committed 2 years ago

01412f76

Deprecate `jax.clear_backends`.

yueshengys committed 2 years ago

147c363e

test digitize corner case and fix it

Gattocrucco committed 2 years ago

fb91b513

Older