nuvec 0.4.1

Pure-Rust array framework for Apple Silicon in the spirit of MLX: lazy n-dimensional arrays (tensors), unified memory, composable transforms
Documentation 
1
2
3
4
5
6
7
8
9
10

//! `checkpoint` stores its closure in the graph: borrowing a local is rejected.
#[path = "stub/mod.rs"]
mod stub;
use stub::*;

fn main() {
    let w = Array;
    let block = checkpoint(|x: &Array| x.matmul(&w));
    let _ = block(&Array);
}