nuvec 0.6.1

Pure-Rust MLX for Apple Silicon: lazy arrays on unified memory, composable transforms and Metal kernels
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
28
29
30
31
32

//! Bounds only where a type parameter is used: `T` visited, `C` hashed, `M` only in
//! `PhantomData` (no bound beyond `'static`).
use std::marker::PhantomData;

use nuvec::{Array, Static, Tree};

#[derive(Clone, Tree)]
struct Generic<T, C, M> {
    item: T,
    #[tree(skip)]
    config: C,
    marker: PhantomData<M>,
}

#[derive(Clone, Static)]
enum Act {
    Relu,
    Gelu { approx: bool },
}

#[derive(Clone)]
struct NotATree;

fn main() {
    let g: Generic<Vec<Array>, Act, NotATree> = Generic {
        item: vec![nuvec::array![1.0]],
        config: Act::Gelu { approx: true },
        marker: PhantomData,
    };
    assert_eq!(g.paths(), ["item.0"]);
    let _ = Act::Relu;
}