pmetal-core 0.1.2

Core types, traits, and configuration for PMetal LLM fine-tuning
Documentation 
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

//! Core types, traits, and configuration for PMetal LLM fine-tuning.
//!
//! This crate provides the foundational abstractions used throughout the PMetal
//! framework, including:
//!
//! - Core trait definitions for models, trainers, and quantizers
//! - Configuration types for training, LoRA, and model loading
//! - Common type definitions (Dtype, Device, etc.)
//! - Error handling infrastructure
//! - Learning rate schedulers
//! - Secure handling of secrets (tokens, credentials)

#![warn(missing_docs)]

mod config;
mod error;
pub mod scheduler;
mod secrets;
mod traits;
mod types;

pub use config::*;
pub use error::*;
pub use scheduler::{LearningRateScheduler, SchedulerBuilder};
pub use secrets::SecretString;
pub use traits::*;
pub use types::*;

/// Prelude module for convenient imports.
pub mod prelude {
    pub use crate::config::*;
    pub use crate::error::{PMetalError, Result};
    pub use crate::scheduler::{LearningRateScheduler, SchedulerBuilder};
    pub use crate::secrets::SecretString;
    pub use crate::traits::*;
    pub use crate::types::*;
}