1 files changed, 4 insertions, 4 deletions
diff --git a/candle-nn/src/lib.rs b/candle-nn/src/lib.rs
index e8086238..45edfc46 100644
--- a/candle-nn/src/lib.rs
+++ b/candle-nn/src/lib.rs
@@ -15,9 +15,9 @@ pub mod vision;
 
 pub use activation::Activation;
 pub use conv::{Conv1d, Conv1dConfig};
-pub use embedding::Embedding;
+pub use embedding::{embedding, Embedding};
 pub use init::Init;
-pub use layer_norm::LayerNorm;
-pub use linear::Linear;
+pub use layer_norm::{layer_norm, LayerNorm};
+pub use linear::{linear, linear_no_bias, Linear};
 pub use optim::SGD;
-pub use var_builder::VarBuilder;
+pub use var_builder::{VarBuilder, VarMap};