ROCm/jax

Pull Requests Commits

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

mminutoli committed 98 days ago

d4d21e6c

[ROCm] Skip AWS credentials and artifact upload on pull_request events

mminutoli committed 98 days ago

5e595fec

[ROCm] Add --copy-props none to s3 sync in promote/publish workflows

mminutoli committed 98 days ago

e060f73c

[ROCm] Fix promote/publish workflows to accept full S3 staging URI

mminutoli committed 98 days ago

860e9760

[ROCm] Add two-mode jax/jaxlib resolution to release pipeline

mminutoli committed 98 days ago

b18a81e6

[ROCm] Add release pipeline CI workflows for ROCm fork

mminutoli committed 98 days ago

05d5653f

[ROCm] Update CI runner labels to amd-do-linux.jax.gpu.gfx950.{1,4,8} (#807)

mminutoli committed 98 days ago

d9029cc0

[ROCm] Put TheRock SDK bin on PATH via so rocminfo/rocm-smi resolve in /opt/rocm-less images

gulsumgudukbay committed 98 days ago

c9faa390

[ROCm] Embed TheRock-aware wheel RUNPATH at link time

gulsumgudukbay committed 98 days ago

34ae6cb1

[ROCm] Add TheRock 7.13 and TheRock latest to CI test matrices

mminutoli committed 98 days ago

658646ba

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

superbobry committed 98 days ago

055a433b

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

allanrenucci committed 98 days ago

425c9c03

Add PDL kernel launch integration to pallas mosaic gpu

rajaanand-g committed 98 days ago

5c65d574

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

hawkinsp committed 98 days ago

f7337b26

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

Google-ML-Automation committed 98 days ago

5f8829ca

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

Google-ML-Automation committed 98 days ago

3673b557

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

Google-ML-Automation committed 98 days ago

4c290b69

Reverts 52e03d80559778007f3d9a379a06ce8f003332d8

brianwa84 committed 98 days ago

ca202035

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

rdyro committed 99 days ago

2f682d05

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

Google-ML-Automation committed 99 days ago

f6237c2d

Skip a ragged_collective_test that is crashing on the released jaxlib.

hawkinsp committed 99 days ago

a723a69b

adjust internal deprecations API.

Jake VanderPlas committed 99 days ago

5e39c370

Fix use of unregistered mlir dialect in autodidax example.

hawkinsp committed 99 days ago

3d24c4a2

add correct ct check to refaccum

mattjj committed 99 days ago

9fc04c36

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

Google-ML-Automation committed 99 days ago

9f842b72

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

tlongeri committed 99 days ago

85425ccf

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

Google-ML-Automation committed 99 days ago

9b6874ae

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

Jake VanderPlas committed 99 days ago

11cefe97

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

Google-ML-Automation committed 99 days ago

59c17f04

Remove fancy batching rule for lax.empty_p

jakevdp committed 99 days ago

967391f0

Older