ds_r1_rs/
lib.rs

1//! # DeepSeek R1 Rust Implementation
2//!
3//! A prototype implementation of DeepSeek R1-inspired reasoning model in Rust.
4//! This library provides core components for transformer architecture, multi-head
5//! latent attention (MLA), mixture-of-experts (MoE), and reasoning capabilities.
6
7pub mod inference;
8pub mod model;
9pub mod training;
10pub mod utils;
11
12// Re-export commonly used types and functions
13pub use model::{config::ModelConfig, transformer::DeepSeekR1Model};
14
15pub use inference::{engine::InferenceEngine, reasoning::ReasoningOutput};
16
17pub use training::{data::TrainingExample, trainer::BasicTrainer};
18
19pub use utils::{
20    error::{ModelError, Result},
21    math::MathUtils,
22};
23
24/// Library version
25pub const VERSION: &str = env!("CARGO_PKG_VERSION");
26
27/// Default model configuration for quick prototyping
28pub fn default_config() -> ModelConfig {
29    ModelConfig::default()
30}
31
32#[cfg(test)]
33mod tests {
34    use super::*;
35
36    #[test]
37    fn test_version() {
38        assert!(!VERSION.is_empty());
39    }
40
41    #[test]
42    fn test_default_config() {
43        let config = default_config();
44        assert!(config.vocab_size > 0);
45        assert!(config.hidden_size > 0);
46        assert!(config.num_layers > 0);
47    }
48}
ds_r1_rs/lib.rs

ds_r1_rs/
lib.rs
