crabml 0.1.0

crabml core package
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

mod add;
mod batch_matmul_vec;
mod div;
mod matmul_vec;
mod mul;
mod rms_norm;
mod rope;
mod silu;
mod softmax;

pub use add::add_inplace;
pub use batch_matmul_vec::batch_matmul_vec;
pub use div::div_inplace;
pub use matmul_vec::matmul_vec;
pub use mul::mul_inplace;
pub use rms_norm::rms_norm_inplace;
pub use rope::rope_inplace;
pub use silu::silu_inplace;
pub use softmax::softmax_inplace;