nuvec 0.4.0

Pure-Rust array framework for Apple Silicon in the spirit of MLX: lazy arrays, unified memory, composable transforms
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

//! The bare-`Array` lint is syntactic: a type alias of `Array` is not caught (documented),
//! and `#[module(..)]` is an exact alias of `#[tree(..)]`.
use nuvec::macros::Module;
use nuvec::{Param, Tree};

type Weights = nuvec::Array;

#[derive(Clone, Module)]
struct Layer {
    weight: Param,
    aliased: Weights,
    #[module(skip)]
    freqs: nuvec::Array,
    #[module(buffer)]
    running: nuvec::Array,
}

fn main() {
    let l = Layer {
        weight: Param::new(nuvec::array![1.0]),
        aliased: nuvec::array![2.0],
        freqs: nuvec::array![3.0],
        running: nuvec::array![4.0],
    };
    assert_eq!(l.paths(), ["weight", "aliased", "running"]);
    let _ = &l.freqs;
}