cubek-attention 0.2.0

CubeK: Attention Kernels
Documentation
1
2
3
4
5
6
mod attention;
mod config;
mod setup;

pub use attention::*;
pub use setup::SimpleBatchAttentionFamily;