cubek-attention 0.2.0

CubeK: Attention Kernels
Documentation
1
2
3
4
5
6
7
8
/// Accelerated but using shared memory for rowwise operations
pub mod blackbox_accelerated;
/// Unit attention
pub mod unit;

mod base;

pub use base::*;