Commits

Commits on Mar 19, 2026

[tokamax] Moving GLU to OSS.

cperivol
authored and
copybara-github
committed
7266f68
[tokamax] Add support for tokamax SM100 attention bias.

cperivol
authored and
copybara-github
committed
0e2dce2

Commits on Feb 23, 2026

[tokamax:attention] sm100 attention and (nondeterministic) vjp kernel

cperivol
authored and
copybara-github
committed
37d2437

Commits on Oct 10, 2025

[tokamax:ragged] Move wgmma_wait(0) to the end of the loop.

cperivol
authored and
copybara-github
committed
2136f9b

Commits on Sep 10, 2025

Rollback of efficient GroupInfo due to breakages.

cperivol
authored and
copybara-github
committed
b32b292

Commits on Aug 28, 2025

[pallas:gpu] Move delay_release to the BlockSpec.

cperivol
authored and
copybara-github
committed
d9c45b9