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 cq;
11pub mod ffn;
12pub mod hadamard;
13pub mod kernels;
14pub mod layers;
15pub mod math;
16pub mod model;
17pub mod norm;
18pub mod ops;
19pub mod quant;
20pub mod rope;
21pub mod v2;
22pub mod v3;
23
24pub use config::TransformerConfig;