2 unstable releases

Uses new Rust 2024

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

#8 in #convolution

MIT/Apache

1.5MB
34K SLoC


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

~56–96MB
~1.5M SLoC