burn_dragon_kernel 0.5.0

Fused GPU kernel crate for burn_dragon execution paths
Documentation 
1
2
3
4
5

pub mod linear;
pub mod mamba;
pub mod mamba2;
pub mod mamba3;
pub mod rwkv8;