"""Token Masking kernel implementations for various backends."""