1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
//! Transformer models.

pub mod albert;

pub mod bert;

mod encoder;
pub use encoder::Encoder;

mod layer_output;
pub use layer_output::{HiddenLayer, LayerOutput};

pub mod roberta;

pub mod sinusoidal;

pub mod squeeze_albert;

pub mod squeeze_bert;

mod traits;