pub fn keyword_lookup(s: &str) -> Option<TokenKind>
Look up keyword/dtype/device from an identifier string. Returns None if the string is a plain identifier.