Training Examples?

[vgoklani]

Are there any examples for training? We would like to use the MoE kernels for gemma4/qwen3.x, do you have any plans on releasing examples for training? I'm asking because I don't see a .backward() method in your MoE kernel.

Thanks