[WIP] Add training kernels#1

emcastillo wants to merge 2 commits intoSakanaAI:masterSakanaAI/sparser-faster-llms:masterfrom

emcastillo:trainingemcastillo/sparser-faster-llms:trainingCopy head branch name to clipboard