1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
//! Model Serialization Module
//!
//! Provides two serialization formats:
//!
//! ## `SafeTensors` Format
//! Industry-standard format compatible with `HuggingFace` ecosystem.
//! ```text
//! [8-byte header: u64 metadata length (little-endian)]
//! [JSON metadata: tensor names, dtypes, shapes, data_offsets]
//! [Raw tensor data: F32 values in little-endian]
//! ```
//!
//! ## APR Format
//! Compact binary format optimized for WASM deployment with JSON metadata.
//! ```text
//! [4-byte magic: "APR1"]
//! [4-byte metadata_len][JSON metadata]
//! [4-byte n_tensors][tensor index][tensor data]
//! [4-byte CRC32]
//! ```
//!
//! Example (SafeTensors):
//! ```rust
//! use aprender::linear_model::LinearRegression;
//! use aprender::primitives::{Matrix, Vector};
//! use aprender::traits::Estimator;
//!
//! # let x = Matrix::from_vec(3, 1, vec![1.0, 2.0, 3.0]).expect("valid matrix dimensions");
//! # let y = Vector::from_vec(vec![2.0, 3.0, 4.0]);
//! let mut model = LinearRegression::new();
//! model.fit(&x, &y).expect("linear regression fit should succeed");
//!
//! // Save to SafeTensors
//! model.save_safetensors("model.safetensors").expect("save safetensors should succeed");
//!
//! // Load from SafeTensors
//! let loaded = LinearRegression::load_safetensors("model.safetensors").expect("load safetensors should succeed");
//! # std::fs::remove_file("model.safetensors").ok();
//! ```
//!
//! Example (APR with metadata):
//! ```rust
//! use aprender::serialization::apr::{AprWriter, AprReader};
//! use serde_json::json;
//!
//! let mut writer = AprWriter::new();
//! writer.set_metadata("model_type", json!("whisper-tiny"));
//! writer.set_metadata("vocab", json!(["hello", "world"]));
//! writer.add_tensor_f32("weights", vec![2, 2], &[1.0, 2.0, 3.0, 4.0]);
//!
//! let bytes = writer.to_bytes().expect("APR writer to_bytes should succeed");
//! let reader = AprReader::from_bytes(bytes).expect("APR reader from_bytes should succeed");
//! assert_eq!(reader.get_metadata("model_type").expect("model_type metadata should exist"), "whisper-tiny");
//! ```
pub use ;
pub use SafeTensorsMetadata;