Skip to main content

forge/
lib.rs

1//! Forge — a WebGPU-native machine learning framework in Rust, scoped to GPT-2.
2//!
3//! Production execution targets WebGPU via `wgpu`; the CPU backend is a
4//! mathematically identical reference used for testing and verification.
5
6pub mod autograd;
7pub mod backend;
8pub mod device;
9pub mod dtype;
10pub mod error;
11pub mod models;
12pub mod nn;
13pub mod ops;
14pub mod optim;
15pub mod serialization;
16pub mod shape;
17pub mod tensor;
18pub mod tokenizer;
19#[cfg(target_arch = "wasm32")]
20pub mod wasm;
21
22pub use device::Device;
23pub use dtype::DType;
24pub use error::{ForgeError, Result};
25pub use models::gpt2::{AttnStep, Gpt2, Gpt2Config, KvCache, LayerDetail, Sampling, StepTrace};
26pub use shape::Shape;
27pub use tensor::Tensor;
28pub use tokenizer::{AnyTokenizer, CharTokenizer, Gpt2Tokenizer, Tokenizer};