Skip to main content

entrenar/optim/simd/
mod.rs

1//! SIMD-accelerated parameter update operations via Trueno
2//!
3//! This module provides vectorized implementations of common optimizer update
4//! operations using Trueno's multi-backend SIMD support. These functions can
5//! provide significant speedup for large parameter tensors.
6
7mod adam;
8mod adamw;
9mod axpy;
10
11#[cfg(test)]
12mod tests;
13
14// Re-export all public functions
15pub use adam::simd_adam_update;
16pub use adamw::simd_adamw_update;
17pub use axpy::simd_axpy;