#tensor #artificial-intelligence #galois

nightly wwml

Tensor library for AI

1 unstable release

new 0.1.0 Mar 26, 2025

#257 in Machine learning


Used in vectorbase

MIT license

315KB
7K SLoC

galois

Tensor library for AI, suitable for AI inference engine development

example

use galois::{arr, cube, mat, Shape, Tensor};
fn main() {
    let m1 = mat(&[
        [0.0, 0.0, 0.0],
        [1.0, 1.0, 1.0],
        [2.0, 2.0, 2.0],
        [3.0, 3.0, 3.0],
    ]);
    let m2 = arr(&[1.0, 2.0, 3.0]);
    let m3 = &m1 + &m2;

    let c = cube(&[[[1.0, 2.0], [3.0, 4.0]], [[5.0, 6.0], [7.0, 8.0]]]);
    let a = arr(&[1.0, 2.0]);

    let c1 = &c + &a;
    let c2 = &c * &a;

    let m1 = Tensor::<f64>::with_shape(
        vec![1.0, 1.0, 1.0, 2.0, 2.0, 2.0],
        Shape::from_slice([2, 3]),
    );
    let m2 = Tensor::<f64>::with_shape(vec![1.0, 2.0, 3.0], Shape::from_slice([3]));
    let m3 = &m1 + &m2;
    let m4 = &m1 * &m2;
}

RUSTFLAGS="-C target-feature=+avx,+avx2,+fma,+f16c -C relocation-model=pic -C link-arg=-pthread -C opt-level=3 -Ctarget-cpu=native" cargo run

Dependencies

~31MB
~712K SLoC