cubek-attention 0.2.0

CubeK: Attention Kernels
Documentation
1
2
3
4
5
6
7
8
9
10
mod attention;
mod config;
mod reader;
mod setup;
mod writer;

pub use attention::*;
pub use reader::*;
pub use setup::SimpleGlobalAttentionFamily;
pub use writer::*;