ghostflow_nn/
lib.rs

1//! GhostFlow Neural Network Layers
2//!
3//! High-level building blocks for neural networks.
4
5pub mod module;
6pub mod linear;
7pub mod conv;
8pub mod norm;
9pub mod activation;
10pub mod dropout;
11pub mod loss;
12pub mod init;
13pub mod attention;
14pub mod transformer;
15pub mod embedding;
16pub mod pooling;
17
18pub use module::Module;
19pub use linear::Linear;
20pub use conv::{Conv1d, Conv2d};
21pub use norm::{BatchNorm1d, BatchNorm2d, LayerNorm};
22pub use activation::*;
23pub use dropout::Dropout;
24pub use loss::*;
25pub use attention::{MultiHeadAttention, scaled_dot_product_attention};
26pub use transformer::{
27    TransformerEncoder, TransformerEncoderLayer,
28    TransformerDecoderLayer, FeedForward,
29    PositionalEncoding, RotaryEmbedding,
30};
31pub use embedding::Embedding;
32pub use pooling::*;
33
34/// Prelude for convenient imports
35pub mod prelude {
36    pub use crate::{Module, Linear, Conv2d, BatchNorm2d, LayerNorm, Dropout};
37    pub use crate::activation::*;
38    pub use crate::loss::*;
39    pub use crate::attention::MultiHeadAttention;
40    pub use crate::transformer::{TransformerEncoder, TransformerEncoderLayer};
41    pub use crate::embedding::Embedding;
42}