Lib.rs

› Science › Machine learning › tract
#linear-algebra #tensorflow #neural-network #onnx #inference #tiny #self-contained

tract-linalg

Tiny, no-nonsense, self contained, TensorFlow and ONNX inference

by Mathieu Poumeyrol and 68 contributors

137 releases

new 0.21.9 Jan 8, 2025
0.21.8 Dec 5, 2024
0.21.7 Sep 23, 2024
0.21.6 Jul 24, 2024
0.2.9 Mar 28, 2019

#670 in Machine learning

Download history 7485/week @ 2024-09-22 6838/week @ 2024-09-29 6984/week @ 2024-10-06 10701/week @ 2024-10-13 9775/week @ 2024-10-20 5473/week @ 2024-10-27 4416/week @ 2024-11-03 2452/week @ 2024-11-10 3811/week @ 2024-11-17 4759/week @ 2024-11-24 3605/week @ 2024-12-01 4963/week @ 2024-12-08 3497/week @ 2024-12-15 1453/week @ 2024-12-22 1888/week @ 2024-12-29 4566/week @ 2025-01-05

11,598 downloads per month
Used in 38 crates (2 directly)

MIT/Apache

1MB
26K SLoC

Rust 17K SLoC // 0.0% comments Templ 9K SLoC // 0.1% comments GNU Style Assembly 13 SLoC // 0.3% comments

tract-linalg

linalg stands for "linear algebra". This is a misnamer. This crates contains low-level, architecture dependant optimisations used by tract-core.

Functions

  • MatMatMul: Extended matrix*matrix product:
    • inspired by Gotoblass and BLIS micro kernel approach
    • extended for convolution friendly addressing (fused img2col)
    • fused output pipeline (min, max, and a few more simple, fast ops)
    • f32*f32 -> f32 (à la sgemm)
    • i8*i8 -> i32 accumulator -> i32 storage
    • i8*i8 -> i32 accumulator -> i8 (with channel zeropoint and scale, and re-quantization pipeline)
  • f32 sigmoid and f32 tanh: at f32 precision, by a rationale function (no exponentiation)
  • byte-to-byte lookup table

Implementations

generic fallback armv6, vfp armv7 neon armv8 simd x64 FMA
MatMatMul f32 4x4 8x4 8x8 16x6
MatMatMul i8->i8 8x4 8x8
MatMatMul i8->i32 8x8
sigmoid f32 4n 4n
tanh f32 4n 4n
byte lookup

Dependencies

~10–18MB
~248K SLoC