xn 0.1.1

Another minimalist deep-learning framework optimized for inference.
Documentation 
1
2
3
4
5
6
7
8
9

pub const ARITHMETIC: &str = include_str!(concat!(env!("OUT_DIR"), "/arithmetic.ptx"));
pub const BROADCAST: &str = include_str!(concat!(env!("OUT_DIR"), "/broadcast.ptx"));
pub const CONV: &str = include_str!(concat!(env!("OUT_DIR"), "/conv.ptx"));
pub const FATTN: &str = include_str!(concat!(env!("OUT_DIR"), "/fattn.ptx"));
pub const FILL: &str = include_str!(concat!(env!("OUT_DIR"), "/fill.ptx"));
pub const INDEXING: &str = include_str!(concat!(env!("OUT_DIR"), "/indexing.ptx"));
pub const LAYOUT: &str = include_str!(concat!(env!("OUT_DIR"), "/layout.ptx"));
pub const REDUCE: &str = include_str!(concat!(env!("OUT_DIR"), "/reduce.ptx"));
pub const ROPE: &str = include_str!(concat!(env!("OUT_DIR"), "/rope.ptx"));