rustorch 0.6.29

Production-ready PyTorch-compatible deep learning library in Rust with special mathematical functions (gamma, Bessel, error functions), statistical distributions, Fourier transforms (FFT/RFFT), matrix decomposition (SVD/QR/LU/eigenvalue), automatic differentiation, neural networks, computer vision transforms, complete GPU acceleration (CUDA/Metal/OpenCL), SIMD optimizations, parallel processing, WebAssembly browser support, comprehensive distributed learning support, and performance validation
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
//! PyTorch model architecture parsing and analysis
//! PyTorchモデルアーキテクチャの解析と分析

pub mod core;
pub mod errors;
pub mod formats;
pub mod types;
pub mod validation;

#[cfg(test)]
pub mod tests;

// Re-export the main types and functions for backward compatibility
pub use core::ModelParser;
pub use errors::{ExecutionResult, ParsingError, ParsingResult};
pub use formats::{ArchitectureDescription, ConnectionDefinition, LayerDefinition, ModelMetadata};
pub use types::{LayerInfo, LayerType, ModelGraph};
pub use validation::ModelValidator;