burn_dragon_hatchling 0.2.0

burn (baby) dragon hatchling inference and training
Documentation
1
2
3
4
5
pub mod block_sparse;
pub mod linear_attention;
pub mod relu_lowrank;

pub use block_sparse::{BlockPattern1d, BlockPattern2d, BlockSparseConfig};