ROCm/jax

Pull Requests Commits

[ROCm]: Add rocm as platform in export_harnessess_multi_platform_test

Rahul Batra committed 1 year ago

11c35cd8

[ROCm]: Add rocm as platform in export_test

Rahul Batra committed 1 year ago

82ee60bc

Update XLA dependency to use revision

a-googler committed 1 year ago

604f5eca

Add option to set coordinator lookup timeout for TPU clusters

a-googler committed 1 year ago

c5869feb

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

a-googler committed 1 year ago

33e1a962

Merge pull request #20323 from jakevdp:fix-doctest

a-googler committed 1 year ago

604cdbcf

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

tlongeri committed 1 year ago

99fadcbc

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

a-googler committed 1 year ago

df9cefab

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

a-googler committed 1 year ago

6fa75aad

doctest: avoid modifying global flag state

jakevdp committed 1 year ago

4963ae16

lax_test: make ComplexPlane sign tests compatible with numpy 2

jakevdp committed 1 year ago

9062cfba

Fix link in the error message.

Jieying Luo committed 1 year ago

cc068360

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

a-googler committed 1 year ago

e6332d3a

[key reuse] add internal function_type_signature utility

jakevdp committed 1 year ago

332450ba

Merge pull request #20314 from jakevdp:dep-maps

a-googler committed 1 year ago

d95df141

Skip offloading tests when used together with an old jaxlib

apaszke committed 1 year ago

e5a16a01

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

a-googler committed 1 year ago

3830b172

Remove internal references to deprecated jax.experimental.maps

jakevdp committed 1 year ago

84e49bd6

Register maps module deprecation outside of module

Jake VanderPlas committed 1 year ago

d8662886

Require array-like inputs to sparse_plus

Jake VanderPlas committed 1 year ago

b48aec57

Strip device_assignment on GPU platform.

a-googler committed 1 year ago

0b28a4b1

Update XLA dependency to use revision

a-googler committed 1 year ago

a4533010

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

hawkinsp committed 1 year ago

f7594522

Remove outdated documentation section,

cheshire committed 1 year ago

0e1b3e5b

[array-api] update to latest array_api_tests commit

jakevdp committed 1 year ago

dad31cf8

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

a-googler committed 1 year ago

cfb968d2

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

a-googler committed 1 year ago

d8f231a8

Merge pull request #19602 from chaserileyroberts:chase/pcollective_broadcast

a-googler committed 1 year ago

d6588ac5

chaserileyroberts committed 1 year ago

01412f76

Deprecate `jax.clear_backends`.

yueshengys committed 1 year ago

147c363e

Older