Module ops 

CUDA tensor operations

Modules

attention

Attention operations (Flash Attention)

conv

Convolution operations

elementwise

Element-wise operations

matmul

Matrix operations

reduction

Reduction operations