ROCm/jax

Pull Requests Commits

[ROCm] Restore download-jax-from-gcs: 0 in blocking-gate (jax built from source)

mminutoli committed 15 days ago

127c03bc

[ROCm] Add skip-jaxlib-from-gcs to download action; run download for build_jaxlib=wheel

mminutoli committed 15 days ago

9eb00308

[ROCm] Switch blocking-gate to build_jaxlib=wheel with S3 GATE_CHECK plugin

mminutoli committed 15 days ago

0db0eac7

[ROCm] Add blocking Bazel PR gate and file-driven test selection

mminutoli committed 15 days ago

1f830c68

[mosaic] Enabled `use-nameloc-as-prefix` for IR dumps

superbobry committed 15 days ago

055a433b

[Mosaic GPU] Add support for concatenating `FragmentedArray`s.

allanrenucci committed 15 days ago

425c9c03

Add PDL kernel launch integration to pallas mosaic gpu

rajaanand-g committed 15 days ago

5c65d574

Fix bug where jnp.int64(1 << 40) would truncate even if explicit x64 types are allowed.

hawkinsp committed 15 days ago

f7337b26

Merge pull request #38694 from aybchan:aybchan/skip-tile-scatter4

Google-ML-Automation committed 15 days ago

5f8829ca

Merge pull request #38724 from vfdev-5:pallas-fix-_gmem_slice_and_predicate

Google-ML-Automation committed 15 days ago

3673b557

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

Google-ML-Automation committed 15 days ago

4c290b69

Reverts 52e03d80559778007f3d9a379a06ce8f003332d8

brianwa84 committed 15 days ago

ca202035

Enable HBM input buffer in `emit_pipeline` (via a simple window slice).

rdyro committed 16 days ago

2f682d05

Merge pull request #38779 from mattjj:ct-check-refaccum

Google-ML-Automation committed 16 days ago

f6237c2d

Skip a ragged_collective_test that is crashing on the released jaxlib.

hawkinsp committed 16 days ago

a723a69b

adjust internal deprecations API.

Jake VanderPlas committed 16 days ago

5e39c370

Fix use of unregistered mlir dialect in autodidax example.

hawkinsp committed 16 days ago

3d24c4a2

add correct ct check to refaccum

mattjj committed 16 days ago

9fc04c36

Merge pull request #38766 from jax-ml:dependabot/github_actions/actions/checkout-7.0.0

Google-ML-Automation committed 16 days ago

9f842b72

[Mosaic][NFC] Add helper function to get the remainder of a Value, if determinable

tlongeri committed 16 days ago

85425ccf

Merge pull request #38587 from Intel-tensorflow:ashahba/rename-oneapi-whl

Google-ML-Automation committed 16 days ago

9b6874ae

re-land #38730: Lower `jnp.empty` & `jnp.empty_like` to `lax.empty`

Jake VanderPlas committed 16 days ago

11cefe97

Merge pull request #38772 from jakevdp:fix-empty-batching

Google-ML-Automation committed 16 days ago

59c17f04

Remove fancy batching rule for lax.empty_p

jakevdp committed 16 days ago

967391f0

[doc] jnp.pad: clarify meaning of `mode="empty"`

Jake VanderPlas committed 16 days ago

ddb1b1d5

If you trace and lower with abstract mesh and abstract inputs and then do `out = lowered.compile(device_assignment=...)(real_arr)`, then make sure the `out.sharding` is concrete i.e. has real devices and is not abstract.

yashk2810 committed 16 days ago

d5bc1dd9

Lower `jnp.empty` & `jnp.empty_like` to `lax.empty`

Google-ML-Automation committed 16 days ago

184e7bf4

Bump actions/checkout from 6.0.3 to 7.0.0

dependabot[bot] committed 16 days ago

Verified 42ef4dff

Merge pull request #38760 from jakevdp:empty-fix

Google-ML-Automation committed 16 days ago

6c1d9dda

Upgrade Abseil to LTS 20260526.0

allanrenucci committed 16 days ago

f027e366

Older