infotheory 1.1.1

The algorithmic information theory library.
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

//! High-performance RWKV7 inference kernel.
//!
//! This module provides a portable SIMD RWKV7 implementation using `wide` so
//! rustc/LLVM can pick the best ISA per target (x86_64, aarch64, wasm32 SIMD,
//! or scalar fallback on non-SIMD CPUs).
//!
//! # Architecture
//!
//! - Matrix/vector operations are vectorized via `wide` (`f32x8`)
//! - State updates are optimized for RWKV7 head dimension N=64
//! - Memory layout is cache-friendly and alignment-aware
//! - No external BLAS dependencies

mod kernel;
mod model;
mod profiling;
mod tensor;
mod weights;

pub use model::ScratchBuffers;
pub use model::{Config, FullAdamState, Model, State, TrainScopeMask};
pub use profiling::{LayerProfiler, LayerTiming, NullProfiler, ProfilerSink};
pub use tensor::{Tensor1D, Tensor2D, TensorView1D, TensorView2D};
pub use weights::Weights;