Skip to content

[Test do not merge] Testing build wheels #722

Closed
alekstheod wants to merge 7513 commits intomainfrom
amd-main
Closed

[Test do not merge] Testing build wheels #722
alekstheod wants to merge 7513 commits intomainfrom
amd-main

Conversation

@alekstheod
Copy link

Test PR

dependabot bot and others added 30 commits February 20, 2026 23:45
Bumps [actions/checkout](https://github.com/actions/checkout) from 6.0.0 to 6.0.1.
- [Release notes](https://github.com/actions/checkout/releases)
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md)
- [Commits](actions/checkout@1af3b93...8e8c483)

---
updated-dependencies:
- dependency-name: actions/checkout
  dependency-version: 6.0.1
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
…ted. Then even if input/output shapes don't match, it should be fine. Fixes jax-ml#35273

PiperOrigin-RevId: 873111177
Co-authored-by: Yash Katariya <yashkatariya@google.com>
* pytrees used_inputs and used_outputs
* the rule can decide what outputs it's going to produce
* don't expose jaxpr eqn, instead just use HiPrimitive instances

Co-authored-by: Yash Katariya <yashkatariya@google.com>
Co-authored-by: Dougal Maclaurin <dougalm@google.com>
Co-authored-by: Yash Katariya <yashkatariya@google.com>
…/actions/checkout-6.0.1

PiperOrigin-RevId: 873964317
These tests require GPU (CUDA or ROCm). Add explicit GPU check
to prevent tests from running on CPU.
The remaining errors require us to bundle more .pyis with jaxlib.
phambinhfin and others added 28 commits March 2, 2026 10:20
Co-authored-by: Daniel Suo <danielsuo@gmail.com>
Co-authored-by: Jake VanderPlas <jakevdp@google.com>
@alekstheod alekstheod requested a review from a team as a code owner March 4, 2026 14:02
@alekstheod alekstheod closed this Mar 4, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.