Amangu/dev#40

Open

aman2930 wants to merge 19 commits intosanbao_tracefrom

Commits on Oct 29, 2025

Adding a benchmark for gemm with input/output bf16 with fp32 accum.
aman2930
committed

Commits on Oct 30, 2025

Commits on Oct 31, 2025

Commits on Nov 4, 2025

Commits on Nov 5, 2025

Replacing tokamax.ragged_dot() with jax.lax.ragged_dot().
aman2930
committed

Commits on Nov 21, 2025

Adding gemm_grouped benchmark configs and fixing gemm_simple config
aman2930
committed