ROCm/jax

Pull Requests Commits

Remove upload-test-artifacts from ROCm workflows

psanal35 committed 2 days ago

cc4c2243

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

Google-ML-Automation committed 2 days ago

2f420124

[ROCm] Pin test_vmap_ellipsis to highest matmul precision

magaonka-amd committed 2 days ago

74b00ca0

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

rdyro committed 2 days ago

5e59322c

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

Google-ML-Automation committed 2 days ago

eca7a3e0

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

Google-ML-Automation committed 2 days ago

75e3d632

Merge pull request #40088 from nouiz:sorted_gpu_info

Google-ML-Automation committed 2 days ago

c66575b3

Merge pull request #40086 from hawkinsp:bazel

Google-ML-Automation committed 3 days ago

a0da1cd8

Update Bazel to v8.7.0.

hawkinsp committed 3 days ago

1d5f3194

Add support for slicing of dynamic shaped memref's including adding support for dynamic tile strides.

WillFroom committed 3 days ago

bb0eb877

Update XLA dependency to use revision http://github.com/openxla/xla/commit/65c8f4341c4bd3c640508e2c8926b6ae989d5232

Google-ML-Automation committed 3 days ago

2dae65cb

Restore native `tpu.logistic` lowering for `jax.nn.silu` and `lax.logistic` on TPU v6+

brianseung-google committed 3 days ago

c853a78e

Match longer GPU name first for good detection

nouiz committed 3 days ago

f8504476

Expose `get_executable_version()` on `PyLoadedExecutable` / `_jax.LoadedExecutable` in JAX.

brianseung-google committed 3 days ago

c3fb2986

Merge pull request #39851 from Intel-tensorflow:kranipa/oneapi_export_test

Google-ML-Automation committed 3 days ago

868f6cb2

Merge pull request #39853 from Intel-tensorflow:kranipa/device_test_oneapi

Google-ML-Automation committed 3 days ago

afc6dbe6

Merge pull request #40079 from jakevdp:jax-export-typing

Google-ML-Automation committed 3 days ago

c2f64619

[typing] use new-style typevars in jax.export

jakevdp committed 3 days ago

940f0524

Handle unimplemented dynamic C-API plugin deserialization gracefully in `aot_test.py`.

brianseung-google committed 3 days ago

49f848dd

Merge pull request #40025 from shashvat-singham:fix/histogram-empty-input

Google-ML-Automation committed 3 days ago

dba480d6

Disable failing Pallas GPU test.

emilyfertig committed 3 days ago

32d906e9

[Pallas:SC] Enable some tests that now work with layout passes

apaszke committed 3 days ago

fd5d80b0

[ROCm] Name ROCm CI jobs after the ROCm release they test

magaonka-amd committed 3 days ago

8221ff4e

[Pallas TPU] Roll back exposing accumulator references to be used with explicit MXU APIs.

bchetioui committed 3 days ago

3e947622

[Pallas/Mosaic GPU] Add a test demonstrating how to perform I/O aliasing.

bchetioui committed 3 days ago

ff4732c1

Merge pull request #40047 from jakevdp:shard-map-typevar

Google-ML-Automation committed 3 days ago

89b3bcac

[typing] use bound typevars in shard_map

jakevdp committed 4 days ago

4d6bf116

Merge pull request #40053 from jax-ml:bzlmod

Google-ML-Automation committed 4 days ago

154945ac

[export] Include disabled_checks= in the error message

superbobry committed 4 days ago

2c5567b4

Update Pyink to Black 26.5.1

AleksMat committed 4 days ago

8a674693

Older