ROCm/jax

Pull Requests Commits

[ROCm] Align hyper parameters

amd-jianli12 committed 68 days ago

237757a4

[ROCm] Support lowering through PJRT_Triton_Extension

amd-jianli12 committed 68 days ago

c0c4c019

[PjRt-IFRT] `ifrt::PjRtArray::pjrt_layout()` uses `nullptr` to indicate a default layout

hyeontaek committed 135 days ago

c1d711a0

Merge pull request #32660 from jax-ml:postrelease-0.8.0

Google-ML-Automation committed 135 days ago

9e7c4e8e

Update CHANGELOG.md

jakeharmon8 committed 135 days ago

Verified 0934e566

Merge branch 'release/0.8.0' into postrelease/0.8.0

jakeharmon8 committed 135 days ago

ba31c4a1

[Mosaic TPU] Support loading and storing float4_e2m1fn

megrez-yliu committed 135 days ago

ff5509b3

Reverts ba9e837026cdcaccb222c7bdd68e2a889fc5b83f

hawkinsp committed 135 days ago

48c87622

Reverts f44bc6e37f770a9d79d24061f7960235a2892407

jakeharmon8 committed 136 days ago

403977d2

Reverts f44bc6e37f770a9d79d24061f7960235a2892407

jakeharmon8 committed 136 days ago

c2cebdc0

Merge pull request #32654 from jakevdp:x64-test-fix

Google-ML-Automation committed 136 days ago

59ae8230

Reverts f93314067cc47bba5d3998409834e675e7384b08

Google-ML-Automation committed 136 days ago

1dbda278

[test] future-proof x64_context_test

jakevdp committed 136 days ago

09c4ff6f

Merge pull request #32649 from jakevdp:cov-dtype

Google-ML-Automation committed 136 days ago

be76f23d

Reverts 1bd3402c63f3c5c0614af1d732389230d18ec237

Google-ML-Automation committed 136 days ago

ba9e8370

jnp.cov & jnp.corrcoef: add support for the dtype argument

jakevdp committed 136 days ago

49f287ff

Merge pull request #32640 from gnecula:fix_back_compat_doc

Google-ML-Automation committed 136 days ago

8e8c3faa

Prepare for JAX release 0.8.0

jakeharmon8 committed 136 days ago

b1ce4c57

[Pallas] Fix 64-bit multiplication emulation in Philox kernel

justinjfu committed 136 days ago

07dad8e8

Update XLA hash for JAX release

jakeharmon8 committed 136 days ago

4f06e7a6

Fix race when dialects were registered in an MLIR context when it was being accessed concurrently.

hawkinsp committed 136 days ago

1bd3402c

Clean up dead code.

brianwa84 committed 136 days ago

b942e894

Remove cusolver_potrf_ffi from the list of covered backwards compatibility targets.

gnecula committed 136 days ago

ac87cbe0

[Pallas:MGPU] Expose multicast TMA stores

apaszke committed 136 days ago

69fd2052

[Pallas:MGPU] Add an all-gather implementation

apaszke committed 136 days ago

8eed7dc5

Add tpu_sc.subcore_barrier primitive.

brianwa84 committed 136 days ago

6aae6f43

Improve the documentation for adding backwards compatibility tests.

gnecula committed 136 days ago

56d12552

Add support for -1 size to ref reshaper, for convenience.

brianwa84 committed 136 days ago

8674d771

[Pallas:MGPU] Add ReducedLayout support to inline_mgpu params

Google-ML-Automation committed 136 days ago

5608cb7c

[pallas:sc] Skip `pl.multiple_of` test on old libtpu

superbobry committed 136 days ago

fff4d58d

Older