rusty_gradients/lib.rs
1// src/lib.rs
2//! # RustyGradients
3//!
4//! `rusty-gradients` - это фреймворк для глубокого обучения на Rust,
5//! вдохновленный PyTorch, с фокусом на простоте, производительности и
6//! реализации современных архитектур, таких как Трансформеры.
7//!
8//! Основной компонент - это `Tensor`, который поддерживает автоматическое
9//! дифференцирование (autograd).
10
11pub mod error;
12pub mod tensor;
13pub mod core;
14pub mod nn;
15pub mod ops;
16pub mod optim;
17pub mod losses;
18
19// --- ИСПРАВЛЕНИЕ: Объявляем наш новый модуль с моделью ---
20pub mod models;
21
22// --- Объявляем наш API для WebAssembly ---
23pub mod wasm_api;
24
25// === NEW: Backend abstraction layer ===
26pub mod backend;
27
28// === NEW: TensorV2 with multi-backend support ===
29pub mod tensor_v2;
30
31// === NEW: Operations V2 with autograd for TensorV2 ===
32pub mod ops_v2;
33
34// === NEW: Model serialization (Safetensors + legacy JSON) ===
35pub mod serialization;
36
37// === NEW: Tokenization (Character, BPE, HuggingFace) ===
38pub mod tokenization;