native_neural_network 0.3.1

Lib no_std Rust for native neural network (.rnn)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

pub mod rnn_api {
    pub use crate::rnn_api::{
        build_f32_into, build_f32_with_input_into, build_f64_into, build_f64_with_input_into,
        read_rnn, train, ReadableRnnFormat, RnnApiError,
    };
}
#[cfg(feature = "publisher-trust-service")]
pub mod trust_service {
    pub use crate::trust_service::*;
}

pub mod modules {
    pub mod activations {
        pub use crate::activations::*;
    }
    pub mod attention {
        pub use crate::attention::*;
    }
    pub mod batching {
        pub use crate::batching::*;
    }
    pub mod beam_search {
        pub use crate::beam_search::*;
    }
    pub mod benchmark {
        pub use crate::benchmark::*;
    }
    pub mod conv3d {
        pub use crate::conv3d::*;
    }
    pub mod conv5d {
        pub use crate::conv5d::*;
    }
    pub mod crypto {
        pub use crate::crypto::*;
    }
    pub mod embeddings {
        pub use crate::embeddings::*;
    }
    pub mod engine {
        pub use crate::engine::*;
    }
    pub mod ffi_api {
        pub use crate::ffi_api::*;
    }
    pub mod gradients {
        pub use crate::gradients::*;
    }
    pub mod inference {
        pub use crate::inference::*;
    }
    pub mod initializers {
        pub use crate::initializers::*;
    }
    pub mod kv_cache {
        pub use crate::kv_cache::*;
    }
    pub mod layers {
        pub use crate::layers::*;
    }
    pub mod lora {
        pub use crate::lora::*;
    }
    pub mod losses {
        pub use crate::losses::*;
    }
    pub mod math {
        pub use crate::math::*;
    }
    pub mod metrics {
        pub use crate::metrics::*;
    }
    pub mod moe {
        pub use crate::moe::*;
    }
    pub mod network {
        pub use crate::network::*;
    }
    pub mod normalization {
        pub use crate::normalization::*;
    }
    pub mod optimizers {
        pub use crate::optimizers::*;
    }
    pub mod parser {
        pub use crate::parser::*;
    }
    pub mod precision {
        pub use crate::precision::*;
    }
    pub mod profiler {
        pub use crate::profiler::*;
    }
    pub mod quantization {
        pub use crate::quantization::*;
    }
    pub mod rope {
        pub use crate::rope::*;
    }
    pub mod runtime {}
    pub mod sampling {
        pub use crate::sampling::*;
    }
    pub mod schedulers {
        pub use crate::schedulers::*;
    }
    pub mod scratch {
        pub use crate::scratch::*;
    }
    pub mod sphere5d {
        pub use crate::sphere5d::*;
    }
    pub mod tensor {
        pub use crate::tensor::*;
    }
    pub mod trainer {
        pub use crate::trainer::*;
    }
    pub mod visualization {
        pub use crate::visualization::*;
    }
}