nuvec 0.7.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

//! The real `checkpoint` stores its closure in the graph: borrowing a local is rejected.
use nuvec::{Array, array, checkpoint};

fn main() {
    let w = array![1.0];
    let block = checkpoint(|x: &Array| x.multiply(&w));
    let _ = block(&array![2.0]);
}