Ipanfilo/port fixes to 212#633

Open

ipanfilo wants to merge 14 commits into

release_v2.12_rocmfrom

ipanfilo/port_fixes_to_212

Commits on Jun 17, 2026

Fix TE loading w/o meta packages (#531)
ipanfilo
committed
[ROCm] fix the bug in hipfied optimized cast tranpose flow that two kernels got run (#545)

wangye805
authored and
ipanfilo
committed
PR545 hot fix for un-addressed reviewer comments (#549)

wangye805
authored and
ipanfilo
committed
Ipanfilo/wheel build action (#529)
ipanfilo
committed
CI: Refactor ROCm CI to use GPU-sized runners and build-only jobs (#528)

leo-automation
authored and
ipanfilo
committed
rocm-ci: scope test container to pod-allocated GPUs (#611)

authored and
ipanfilo
committed
[ROCm] add the bias all row -inf support for jax unfused-attn (#556)

authored and
ipanfilo
committed
Disable all UB layer tests for gfx942 (#567)

alextmagro
authored and
ipanfilo
committed
upgrade hypothesis/setuptools (#572)

matthiasdiener
authored and
ipanfilo
committed
Install: use setuptools bdist_wheel; CI: call pytest as module (#601)
ipanfilo
committed
JAX 0.9 compatibility changes (#604)
ipanfilo
committed
Hotfix for Maxtext regression with JAX 0.9 changes (PR#629)
ipanfilo
committed
Add test (PR#629)
ipanfilo
committed
Remove wrong submodule reference
ipanfilo
committed