Split lowering rules out of jax_primitives.py#2753
Open
paul0403 wants to merge 7 commits into
Open
Codecov / codecov/patch
succeeded
Jun 8, 2026 in 1s
97.33% of diff hit (target 96.99%)
View this Pull Request on Codecov
97.33% of diff hit (target 96.99%)
Loading