Re-exports§
pub use attention::MultiHeadAttention;pub use attention::MultiHeadAttentionConfig;pub use baseline::BaselineTransformer;pub use baseline::BaselineTransformerBlock;pub use benchmark::BenchmarkCase;pub use benchmark::BenchmarkComparisonReport;pub use benchmark::BenchmarkConfig;pub use benchmark::BenchmarkDelta;pub use benchmark::BenchmarkError;pub use benchmark::BenchmarkGateOutcome;pub use benchmark::BenchmarkMemory;pub use benchmark::BenchmarkRegressionThresholds;pub use benchmark::BenchmarkReport;pub use benchmark::BenchmarkSuite;pub use benchmark::BenchmarkTarget;pub use benchmark::BenchmarkTiming;pub use benchmark::CompressorBenchmarks;pub use benchmark::DeltaOperatorBenchmarks;pub use benchmark::DeltaResBlockBenchmarks;pub use benchmark::ModelBenchmark;pub use benchmark::ModelBenchmarkComparison;pub use benchmark::ModelBenchmarkSummary;pub use benchmark::ModelMemoryBreakdown;pub use benchmark::NormalizationBenchmarks;pub use benchmark::compare_benchmark_reports;pub use benchmark::run_benchmarks;pub use beta_branch::BetaBranch;pub use checkpoint::CheckpointError;pub use checkpoint::CheckpointManifest;pub use checkpoint::LoadedTrainingArtifact;pub use checkpoint::load_checkpoint;pub use checkpoint::load_manifest;pub use checkpoint::load_training_artifact;pub use checkpoint::load_training_comparison_report;pub use checkpoint::load_training_report;pub use checkpoint::save_checkpoint;pub use checkpoint::save_training_artifact;pub use checkpoint::save_training_comparison_report;pub use checkpoint::save_training_report;pub use checkpoint::save_training_sweep;pub use compressor::ChannelConvCompressor;pub use compressor::TokenConvCompressor;pub use config::CompressionVariant;pub use config::DdlConfig;pub use config::DdlMapping;pub use data::TailStrategy;pub use data::TokenBatch;pub use data::TokenBatcher;pub use data::TokenBatchingConfig;pub use data::TokenDataset;pub use data::TokenDatasetSummary;pub use delta_operator::DeltaOperator;pub use delta_res_block::DeltaResBlock;pub use embed_expander::EmbeddingExpander;pub use error::BaselineConfigError;pub use error::ConfigValidationError;pub use error::DataValidationError;pub use error::SpectralError;pub use eval::ComparisonReport;pub use eval::DatasetComparisonReport;pub use eval::DatasetTiming;pub use eval::DatasetVariantEvaluation;pub use eval::EvalError;pub use eval::VariantEvaluation;pub use eval::compare_variants;pub use eval::compare_variants_on_dataset;pub use eval::compare_variants_on_dataset_with_diagnostics;pub use eval::compare_variants_with_diagnostics;pub use eval::evaluate_variant;pub use eval::evaluate_variant_on_dataset;pub use eval::evaluate_variant_on_dataset_with_diagnostics;pub use eval::evaluate_variant_with_diagnostics;pub use generation::AutoregressiveModel;pub use generation::FinishReason;pub use generation::GenerationConfig;pub use generation::GenerationError;pub use generation::GenerationResult;pub use generation::GenerationStep;pub use generation::generate_tokens;pub use k_branch::KMapBranch;pub use k_branch::VMapKBranch;pub use lm::CausalLmLossSummary;pub use lm::CausalLmMetrics;pub use lm::aggregate_causal_lm_summaries;pub use lm::causal_language_model_loss;pub use lm::causal_language_model_metrics;pub use lm::causal_language_model_metrics_with_lengths;pub use lm::causal_language_model_summary;pub use lm::causal_language_model_summary_with_lengths;pub use mlp::SwiGluMlp;pub use mlp::SwiGluMlpConfig;pub use spectral::BetaHistogram;pub use spectral::DeltaRegime;pub use spectral::DeltaResInfo;pub use spectral::LayerDiagnostics;pub use spectral::LayerSpectralInfo;pub use spectral::ModelDiagnostics;pub use spectral::SpectralCollector;pub use spectral::SpectralDiagnostics;pub use spectral::SpectralInfo;pub use spectral::Sublayer;pub use training::CosineWarmupSchedule;pub use training::TrainingComparisonReport;pub use training::TrainingConfig;pub use training::TrainingError;pub use training::TrainingOutcome;pub use training::TrainingReport;pub use training::TrainingSpectralSnapshot;pub use training::TrainingStepMetrics;pub use training::TrainingSweepOutcome;pub use training::resume_training;pub use training::train_variant;pub use training::train_variants;pub use transformer::DdlTransformer;pub use transformer::DdlTransformerBlock;pub use v_branch::MatrixValueBranch;pub use v_branch::ScalarValueBranch;pub use variant::DiagnosticLevel;pub use variant::ModelInstance;pub use variant::ModelOutput;pub use variant::ModelVariant;