ferrite/lib.rs
1//! Ferrite - A lightweight deep learning framework
2//!
3//! This crate provides a dynamic computation graph with automatic differentiation,
4//! designed for building and training neural networks.
5
6#![allow(unreachable_patterns)]
7#![allow(unused_variables)]
8#![allow(dead_code)]
9
10mod autograd;
11mod tensor;
12mod network;
13
14// Import and re-export macros globally
15#[macro_use]
16mod macros;
17
18// Re-export the main types
19pub use tensor::*;
20pub use network::*;
21pub use autograd::*;
22
23// Version of the crate
24pub const VERSION: &str = env!("CARGO_PKG_VERSION");
25
26// Optional prelude module for convenient imports
27pub mod prelude {
28 pub use crate::tensor::*;
29 pub use crate::autograd::*;
30 pub use crate::network::*;
31 pub use crate::layer; // Re-export the macro
32 pub use Layer::Module;
33
34}