ROCm/jax

Pull Requests Commits

[ROCm] Enable buffer_callback_test_gpu in ROCm CI

magaonka-amd committed 2 days ago

ed8289a7

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

magaonka-amd committed 2 days ago

b108cf2f

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

magaonka-amd committed 2 days ago

4cd5d8be

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

mminutoli committed 2 days ago

ed9ecaff

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

magaonka-amd committed 2 days ago

3b5d9fce

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

Google-ML-Automation committed 2 days ago

019717d4

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

Google-ML-Automation committed 2 days ago

94018b42

Merge pull request #39975 from skye-ant:mtls

Google-ML-Automation committed 2 days ago

c3ca80f8

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

Google-ML-Automation committed 2 days ago

e9c0e74b

Add mTLS support to `jax.distributed` coordination service

skye-ant committed 2 days ago

Verified ebfefa17

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

Google-ML-Automation committed 2 days ago

10f85c02

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

dependabot[bot] committed 2 days ago

Verified faf9e7b2

Add libtpu version guard to fix CI failure.

hawkinsp committed 2 days ago

3407b235

Fix test failures under LAPACK 3.12.1.

hawkinsp committed 2 days ago

97b969ed

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

Google-ML-Automation committed 2 days ago

9b85161c

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

hawkinsp committed 2 days ago

97de27c6

[Mosaic] Support lax.reduce_precision lowering.

WindQAQ committed 2 days ago

c8df3645

PR #40006: [shape_poly] Add symbolic_dim_bounds API

mgoldfarb-nvidia committed 2 days ago

a1ed2bcd

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

Google-ML-Automation committed 2 days ago

194c141b

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

Google-ML-Automation committed 2 days ago

4b48cd48

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

Google-ML-Automation committed 2 days ago

461702c3

Remove upload-test-artifacts from ROCm workflows

psanal35 committed 2 days ago

cc4c2243

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

magaonka-amd committed 2 days ago

d572293b

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

Google-ML-Automation committed 3 days ago

2f420124

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

jakevdp committed 3 days ago

ccbd8611

[ROCm] Pin test_vmap_ellipsis to highest matmul precision

magaonka-amd committed 3 days ago

74b00ca0

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

rdyro committed 3 days ago

5e59322c

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

Google-ML-Automation committed 3 days ago

eca7a3e0

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

Google-ML-Automation committed 3 days ago

75e3d632

Merge pull request #40088 from nouiz:sorted_gpu_info

Google-ML-Automation committed 3 days ago

c66575b3

Older