needle_core/lib.rs
1#![cfg_attr(not(feature = "std"), no_std)]
2#![allow(clippy::excessive_precision)]
3
4extern crate alloc;
5#[cfg(feature = "std")]
6extern crate std;
7
8pub mod attn;
9pub mod config;
10pub mod ffn;
11pub mod layers;
12pub mod math;
13pub mod model;
14pub mod norm;
15pub mod ops;
16pub mod quant;
17pub mod rope;
18
19pub use config::TransformerConfig;