6d8ba0708c
- mamba/mod.rs: ~90 errors fixed — _candle suffixed functions replaced, operator overloads→free functions, autograd→pseudo-gradients, checkpoint→JSON serialization - gpu_tensor.rs: added gpu_eye, gpu_cat_dim0, gpu_stack_tensors - TFT/SSD: unused imports cleaned, type mismatches fixed - ml-core: GpuTensor algebra methods (17 new), cuda_compat.rs deleted, GpuVarStore::vars/all_vars/linear_xavier added Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
35 lines
1.5 KiB
Rust
35 lines
1.5 KiB
Rust
//! Supervised ML models for Foxhunt HFT trading
|
|
//!
|
|
//! Contains 8 model architectures: TFT, Mamba-2, Liquid, TGGN, TLOB, KAN, xLSTM, Diffusion.
|
|
//! Each model directory contains the core network, layers, and self-contained training logic.
|
|
//! Checkpointable impls live in the `checkpoint` module. `UnifiedTrainable` adapters live in `ml`.
|
|
|
|
#![allow(clippy::module_name_repetitions)]
|
|
#![allow(clippy::integer_division)]
|
|
#![allow(clippy::shadow_reuse, clippy::shadow_same, clippy::shadow_unrelated)] // Tensor ops: let x = x.relu() is idiomatic
|
|
#![allow(clippy::non_ascii_literal)] // Math symbols in ML documentation and error messages
|
|
#![allow(clippy::partial_pub_fields)] // ML config structs: some fields are pub API, some internal
|
|
#![allow(clippy::same_name_method)] // Intentional: inherent methods shadow trait defaults for ML-specific behavior
|
|
#![allow(clippy::indexing_slicing)] // Tensor/matrix indexing with bounds guaranteed by construction
|
|
#![allow(clippy::similar_names)] // ML naming: min_val/max_val, state/states are conventional
|
|
#![allow(clippy::type_complexity)] // Complex generic types in neural network layers
|
|
#![allow(clippy::single_char_lifetime_names)] // 'a is idiomatic Rust
|
|
|
|
// Re-export shared types from ml-core
|
|
pub use ml_core::xavier_init;
|
|
|
|
// GPU tensor abstraction (cudarc CudaSlice + cuBLAS)
|
|
pub mod gpu_tensor;
|
|
|
|
pub mod tft;
|
|
pub mod mamba;
|
|
pub mod liquid;
|
|
pub mod tgnn;
|
|
pub mod tlob;
|
|
pub mod kan;
|
|
pub mod xlstm;
|
|
pub mod diffusion;
|
|
|
|
// Checkpoint impls (Checkpointable for Mamba2SSM, TGGN)
|
|
pub mod checkpoint;
|