Skip to main content

Crate axonml

Crate axonml 

Source
Expand description

axonml library root

§File

crates/axonml/src/lib.rs

§Author

Andrew Jewell Sr - AutomataNexus

§Updated

March 8, 2026

§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 monitor::TrainingMonitor;
pub use adversarial::AdversarialTrainer;
pub use adversarial::adversarial_training_step;
pub use adversarial::fgsm_attack;
pub use adversarial::pgd_attack;
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 hub::compare_benchmarks;
pub use hub::list_all_models;
pub use hub::models_by_category;
pub use hub::models_by_max_params;
pub use hub::models_by_max_size_mb;
pub use hub::recommended_models;
pub use hub::search_models;
pub use benchmark::MemorySnapshot;
pub use benchmark::ThroughputConfig;
pub use benchmark::ThroughputResult;
pub use benchmark::print_throughput_results;
pub use benchmark::benchmark_model;
pub use benchmark::benchmark_model_named;
pub use benchmark::compare_models;
pub use benchmark::throughput_test;
pub use benchmark::warmup_model;
pub use benchmark::print_memory_profile;
pub use benchmark::profile_model_memory;
pub use axonml_core as core;
pub use axonml_tensor as tensor;
pub use axonml_autograd as autograd;
pub use axonml_nn as nn;
pub use axonml_optim as optim;
pub use axonml_data as data;
pub use axonml_vision as vision;
pub use axonml_text as text;
pub use axonml_audio as audio;
pub use axonml_distributed as distributed;
pub use axonml_profile as profile;
pub use axonml_llm as llm;
pub use axonml_jit as jit;
pub use axonml_onnx as onnx;

Modules§

adversarial
Adversarial training utilities (FGSM, PGD, adversarial trainer). Adversarial Training Utilities
benchmark
Model Benchmarking Utilities
hub
Unified Model Hub - Central Registry for All Pretrained Models
hvac
HVAC 8-model diagnostic system (~8.6M total parameters). HVAC 8-Model Diagnostic System
monitor
Live browser-based training monitor — opens Chromium with real-time charts. Training Monitor — live browser-based training dashboard
prelude
Common imports for machine learning tasks.
trainer
High-Level Training Utilities

Functions§

features
Returns a string describing the enabled features.
version
Returns the version of the Axonml framework.