burn-gdn2 0.1.1

Gated DeltaNet 2 — linear recurrent token mixer with channel-wise erase/write gates
Documentation
1
2
3
4
5
6
7
8
9
10
11
pub mod fused_recurrent;

// Experimental cubecl kernels — require cubecl >= 0.11
// #[cfg(feature = "cubecl")]
// pub mod cubecl_fused;
// #[cfg(feature = "cubecl")]
// pub mod cubecl_chunk;
// #[cfg(feature = "cubecl")]
// pub mod cubecl_dispatch;
// #[cfg(feature = "cubecl")]
// pub mod cubecl_forward;