[JAX] gemm grouped#133

Draft

mariogeiger wants to merge 23 commits intomainfrom

jax-gemm-grouped

Commits on Jun 26, 2025

add experimental gemm-grouped suport
mariogeiger
committed

Commits on Jul 1, 2025

Add precision parameter to segmented polynomial system
mariogeiger
committed

Commits on Jul 2, 2025

fix
mariogeiger
committed

Commits on Jul 14, 2025

Merge branch 'main' into gemm-grouped-and-method-kwarg
mariogeiger
committed

Commits on Jul 16, 2025

Commits on Oct 22, 2025