tracel-ai / cubecl

Multi-platform high-performance compute language extension for Rust.