mlx-rs 0.33.0

Unofficial rust wrapper for Apple's mlx machine learning library.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

/// See `assertEqual` in the swift binding tests
#[allow(unused_macros)]
macro_rules! assert_array_all_close {
    ($a:tt, $b:tt) => {
        let _b: Array = $b.into();
        let assert = $a.all_close(&_b, None, None, None).unwrap();
        assert!(assert);
    };
}

#[allow(unused_macros)]
macro_rules! cfg_safetensors {
    ($($item:item)*) => {
        $(
            #[cfg(feature = "safetensors")]
            $item
        )*
    };
}