Crate axonml_train 

Axonml Train — High-Level Training Infrastructure

File

crates/axonml-train/src/lib.rs

Author

Andrew Jewell Sr. — AutomataNexus LLC ORCID: 0009-0005-2158-7060

Updated

April 14, 2026 11:15 PM EST

Overview

Training utilities split out of the axonml umbrella crate:

Module	Purpose
trainer	TrainingConfig, Callback, EarlyStopping, ProgressLogger, TrainingHistory, TrainingMetrics, clip_grad_norm
hub	Unified model hub / registry (UnifiedModelInfo, BenchmarkResult, ModelCategory) — requires vision + llm features for full catalog
benchmark	benchmark_model, throughput_test, profile_model_memory, compare_models
adversarial	AdversarialTrainer, fgsm_attack, pgd_attack

The live browser training monitor stays in the axonml umbrella crate (see axonml::monitor::TrainingMonitor).

Feature Flags

• vision — include axonml-vision models in the hub registry
• llm — include axonml-llm models in the hub registry
• full — both vision + llm
• cuda — GPU acceleration

Disclaimer

Use at own risk. This software is provided “as is”, without warranty of any kind, express or implied. The author and AutomataNexus shall not be held liable for any damages arising from the use of this software.

Re-exports

pub use trainer::Callback;

pub use trainer::EarlyStopping;

pub use trainer::ProgressLogger;

pub use trainer::TrainingConfig;

pub use trainer::TrainingHistory;

pub use trainer::TrainingMetrics;

pub use trainer::TrainingState;

pub use trainer::clip_grad_norm;

pub use trainer::compute_accuracy;

pub use hub::BenchmarkResult;

pub use hub::ModelCategory;

pub use hub::UnifiedModelInfo;

pub use benchmark::MemorySnapshot;

pub use benchmark::ThroughputConfig;

pub use benchmark::ThroughputResult;

pub use benchmark::benchmark_model;

pub use benchmark::benchmark_model_named;

pub use benchmark::compare_models;

pub use benchmark::print_memory_profile;

pub use benchmark::print_throughput_results;

pub use benchmark::profile_model_memory;

pub use benchmark::throughput_test;

pub use benchmark::warmup_model;

pub use adversarial::AdversarialTrainer;

pub use adversarial::adversarial_training_step;

pub use adversarial::fgsm_attack;

pub use adversarial::pgd_attack;

Modules

adversarial

Adversarial Training Utilities

benchmark

Model Benchmarking Utilities

hub

Unified Model Hub - Central Registry for All Pretrained Models

trainer

High-Level Training Utilities