ROCm/jax

Pull Requests Commits

Set HSA_NO_SCRATCH_RECLAIM=0 for the ROCm pytest runs

magaonka-amd committed 23 days ago

0b7609a9

[ROCm] Enable buffer_callback_test_gpu in ROCm CI

magaonka-amd committed 24 days ago

ed8289a7

[ROCm] Enable the 35 device-agnostic bazel tests in ROCm CI

magaonka-amd committed 24 days ago

b108cf2f

[ROCm] continuous: build jax/jaxlib on fork runners

magaonka-amd committed 24 days ago

4cd5d8be

[ROCm] nightly: build jax/jaxlib on fork runners

mminutoli committed 24 days ago

ed9ecaff

Enable additional single-GPU bazel tests in ROCm CI (#816)

magaonka-amd committed 24 days ago

3b5d9fce

Merge pull request #40098 from ROCm:rocm-drop-upload-test-artifacts

Google-ML-Automation committed 24 days ago

019717d4

Merge pull request #39578 from Whning0513:docs/update-pprof-command-24407

Google-ML-Automation committed 24 days ago

94018b42

Merge pull request #39975 from skye-ant:mtls

Google-ML-Automation committed 24 days ago

c3ca80f8

Merge pull request #40107 from jax-ml:dependabot/github_actions/astral-sh/setup-uv-10.0.0

Google-ML-Automation committed 24 days ago

e9c0e74b

Add mTLS support to `jax.distributed` coordination service

skye-ant committed 24 days ago

Verified ebfefa17

Merge pull request #39547 from aybchan:add-rubin-gpuinfo

Google-ML-Automation committed 24 days ago

10f85c02

Bump astral-sh/setup-uv from 9.0.0 to 10.0.0

dependabot[bot] committed 24 days ago

Verified faf9e7b2

Add libtpu version guard to fix CI failure.

hawkinsp committed 24 days ago

3407b235

Fix test failures under LAPACK 3.12.1.

hawkinsp committed 24 days ago

97b969ed

Merge pull request #40097 from ROCm:fix/pallas-io-aliasing-test-size-rocm

Google-ML-Automation committed 24 days ago

9b85161c

Fully qualify references to InconclusiveDimensionOperation in the docs to fix ambiguous cross references.

hawkinsp committed 24 days ago

97de27c6

[Mosaic] Support lax.reduce_precision lowering.

WindQAQ committed 24 days ago

c8df3645

PR #40006: [shape_poly] Add symbolic_dim_bounds API

mgoldfarb-nvidia committed 24 days ago

a1ed2bcd

Update XLA dependency to use revision http://github.com/openxla/xla/commit/454e3787563d72a5d0174d92d450c37efc6172b2

Google-ML-Automation committed 24 days ago

194c141b

[Pallas] Enable `round` lowering for SC Vector Subcore.

Google-ML-Automation committed 25 days ago

4b48cd48

Merge pull request #40094 from jakevdp:pyrefly-12-errors

Google-ML-Automation committed 25 days ago

461702c3

Remove upload-test-artifacts from ROCm workflows

psanal35 committed 25 days ago

cc4c2243

[ROCm] Reduce pallas input/output aliasing test size on ROCm

magaonka-amd committed 25 days ago

d572293b

Merge pull request #39974 from ROCm:fix/hijax-einsum-tf32-inputs

Google-ML-Automation committed 25 days ago

2f420124

[typing] fix type checking errors under pyrefly v1.2.0

jakevdp committed 25 days ago

ccbd8611

[ROCm] Pin test_vmap_ellipsis to highest matmul precision

magaonka-amd committed 25 days ago

74b00ca0

[fuser] Update `dynamic_update_slice` rule to return `no_block_spec` for operand.

rdyro committed 25 days ago

5e59322c

Merge pull request #40029 from magaonka-amd:rocm-ci-bazel-test-jax-base-image

Google-ML-Automation committed 25 days ago

eca7a3e0

Merge pull request #39970 from magaonka-amd:cleanup/rocm-ci-job-naming-n-tracks

Google-ML-Automation committed 25 days ago

75e3d632

Older