tokitai_operator/lib.rs
1//! Tokitai operator library: a mathematical operator compiler.
2//!
3//! Tokitai treats algebraic laws, valuation, locality, and precision as
4//! first-class scheduling inputs. The crate is organized as 8 layers:
5//!
6//! 1. [`facade`] — the public `Tokitai` handle, the only entry point a
7//! user needs.
8//! 2. [`object`] — `Tensor`, `ObjectMeta`, `Shape`, `Representation`.
9//! 3. [`ir`] — `SemanticGraph` and the closure-based DSL.
10//! 4. [`planner`] — `HeuristicPlanner`, `LoweringRule`, obligations.
11//! 5. [`op`] — `Operator` trait, `OpSignature`, and the per-family op
12//! modules (`arithmetic`, `shape`, `nn`, `index`, `reductions`).
13//! 6. [`domain`] — `DomainId`, `Contract`, `Claim`, `Evidence`.
14//! 7. [`backend`] — `CpuScalarBackend` (default), `GpuScaffoldBackend`,
15//! `hip_*` (gated on `rocm-hip`).
16//! 8. [`verify`] — `audit_report`, `release_gate`, `claim_status`,
17//! `support_matrix`. This is the source-of-truth for the paper
18//! claim boundary.
19//!
20//! The default feature set is CPU-only and includes all op families,
21//! all planner policies, and the full verifier surface. The `rocm-hip`
22//! feature (opt-in) adds the `hip_*` backends, the 0.7B MoE training
23//! project, and the model server.
24//!
25//! For the architecture overview and the per-claim support matrix, see
26//! `ARCHITECTURE.md`. For the claim-boundary code, see
27//! `src/verify/mod.rs`. For the public surface index, see
28//! `docs/operators.md`.
29//!
30pub mod backend;
31#[cfg(feature = "rocm-hip")]
32pub mod checkpoint;
33pub mod dataset_bridge;
34pub mod domain;
35pub mod error;
36pub mod facade;
37#[cfg(feature = "rocm-hip")]
38pub mod infer;
39pub mod ir;
40pub mod metrics;
41#[cfg(feature = "rocm-hip")]
42pub mod model;
43#[cfg(feature = "rocm-hip")]
44pub mod model_arch;
45#[cfg(feature = "rocm-hip")]
46pub mod moe_model;
47pub mod object;
48pub mod op;
49pub mod planner;
50#[cfg(feature = "rocm-hip")]
51pub mod synth_data;
52pub mod theory;
53#[cfg(feature = "rocm-hip")]
54pub mod training;
55#[cfg(feature = "rocm-hip")]
56pub mod training_runner;
57pub mod verify;
58
59pub use error::{Error, Result};