entrenar 0.7.12

Training & Optimization library with autograd, LoRA, quantization, and model merging
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

//! Compute Device Abstraction (ENT-008)
//!
//! Provides hardware device detection and abstraction for CPU, GPU, TPU,
//! and Apple Silicon devices.

mod apple;
mod compute;
mod cpu;
mod gpu;
mod simd;
mod tpu;

#[cfg(test)]
mod tests;

// Re-export all public types for API compatibility
pub use apple::AppleSiliconInfo;
pub use compute::ComputeDevice;
pub use cpu::CpuInfo;
pub use gpu::GpuInfo;
pub use simd::SimdCapability;
pub use tpu::TpuInfo;