nuvec 0.6.0

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

//! `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);
}