Skip to main content

uqa_ml/
lib.rs

1//
2// Unified Query Algebra
3//
4// Copyright (c) 2023-2026 Cognica, Inc.
5//
6
7//! ML model specs, analytical training, and inference backends.
8
9pub mod backend;
10pub mod deep_fusion;
11pub mod model;
12pub mod training;
13
14#[cfg(feature = "mlx")]
15pub mod mlx;
16
17pub use backend::{CPUBackend, MLBackend, MLError, MLResult};
18pub use deep_fusion::{
19    AggregationKind as DeepAggKind, DeepFusionOperator, Gating, GlobalPoolMethod, Layer,
20    PoolMethod as DeepPoolMethod,
21};
22pub use model::{
23    AggregationSpec, DeepLayerSpec, DeepModel, DirectionSpec, GatingSpec, PoolKindSpec,
24    PoolMethodSpec, PredictResult,
25};
26pub use training::{
27    deep_learn, DeepLearnOutput, LearnOptions, TrainingExample, TrainingReport, TrainingSet,
28};
29
30#[cfg(feature = "mlx")]
31pub use mlx::MLXBackend;