trustformers-core 0.1.1

Core traits and utilities for TrustformeRS
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
// Copyright (c) 2025-2026 COOLJAPAN OU (Team KitaSan)
// SPDX-License-Identifier: Apache-2.0

//! SIMD-optimized operations for performance-critical kernels
//!
//! This module provides SIMD implementations for various neural network operations
//! including layer normalization, softmax, and matrix operations, optimized for
//! different instruction sets (AVX-512, AVX2, NEON, RISC-V Vector).

pub mod cpu_features;
pub mod layer_norm;
pub mod matrix_ops;
pub mod softmax;

pub use cpu_features::CpuFeatures;
pub use layer_norm::SIMDLayerNorm;
pub use matrix_ops::SIMDMatrixOps;
pub use softmax::SIMDSoftmax;