cubecl

CubeCL 是一个面向 Rust 的多平台高性能 GPU 计算语言扩展，用于编写可跨平台部署的零开销抽象计算内核。它支持 CUDA、ROCm、WebGPU、Metal、Vulkan 和 CPU 后端，提供统一的 Cube 编程模型、自动内存管理、懒执行调度及针对张量运算优化的运行时；核心特性包括平台感知编译时检查、SIMD 自动向量化、类型安全的浮点/整数泛型算子。适用于机器学习推理、科学计算、图像处理等需在异构硬件上高效执行数值密集型任务的场景。