burn-nn 0.20.1

Neural network building blocks for the Burn deep learning framework
Documentation 
1
2
3
4
5
6
7

mod cross_attention;
mod mask;
mod mha;

pub use cross_attention::*;
pub use mask::*;
pub use mha::*;