2 unstable releases

Uses new Rust 2024

new 0.1.0-pre.1 Dec 18, 2025
0.0.1 Dec 5, 2025

#4 in #cube-k


Used in cubek

MIT/Apache

36KB
814 lines


Discord Current Crates.io Version Minimum Supported Rust Version Test Status license


CubeK: high-performance multi-platform kernels in CubeCL

Algorithms

Algorithms Variants
Random bernoulli normal uniform
Quantization symmetric per-block per-tensor q2 q4 q8 fp4
Reduction mean sum prod max min arg[max|min] per-cube per-plane
Matmul mma unit tma multi-stage specialization ordered multi-rows
Convolution mma unit tma multi-stage im2col
Attention mma unit multi-rows

Dependencies

~14–50MB
~700K SLoC