vkml 0.0.3

High-level Vulkan-based machine learning library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

//! VKML - High-level abstractions for ML model development using Vulkan compute
//!
//! This library provides universal compute utilisation across different hardware vendors
//! with a focus on performance and ease of use.

mod utils;

mod slang;

mod gpu;

mod compute;
mod scheduler;

mod model;

mod layer;

mod tensor;

mod instruction;

mod tensor_graph;

mod importers;

mod weight_initialiser;

pub use compute::compute_manager::ComputeManager;
pub use compute::optimisations::Optimisations;
pub use importers::onnx_parser;
pub use layer::factory::Layers;
pub use model::{graph_model::GraphModel, layer_connection::LayerConnection};
pub use onnx_extractor::DataType;
pub use tensor::Tensor;
pub use tensor::TensorDesc;
pub use utils::error::VKMLError;