candle-ug 0.10.2

Minimalist ML framework.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

//! This crate is used to re-export the `ug` crate together with `ug-cuda` & `ug-metal` gated
//! behind the `cuda` and `metal` features respectively.

pub use ug::*;

#[cfg(feature = "cuda")]
pub mod cuda {
    pub use ug_cuda::*;
}

#[cfg(feature = "metal")]
pub mod metal {
    pub use ug_metal::*;
}