Kernels:

phanerozoic
/

grouped-gemm-moe

Kernel card Files Files and versions

grouped-gemm-moe / tests

5 kB

Ctrl+K

Ctrl+K

2 contributors

History: 2 commits

CharlesCNorton

Add fused MoE backward: trainable end to end (dx, dW, routing-weight grads)

367d6ba 25 days ago

test_grouped_gemm_moe.py
5 kB
Add fused MoE backward: trainable end to end (dx, dW, routing-weight grads) 25 days ago