rten-vecmath 0.22.0

SIMD vectorized implementations of various math functions used in ML models
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
# rten-vecmath

This crate contains SIMD-vectorized kernels ("vectorized math") for various
operations used in machine learning models. This includes:

 - Math functions such as exp, erf, tanh
 - Activation function such as gelu
 - Normalization functions such as softmax and mean-variance normalization
 - Reduction functions such as sums and sum-of-square

SIMD operations are implemented using portable SIMD types from the rten-simd
crate.