#![allow(deprecated)]
#![allow(unreachable_code)]
#![allow(unused_mut)]
#![allow(unused_parens)]
#![allow(for_loops_over_fallibles)]
#![allow(unexpected_cfgs)]
#![allow(unused_attributes)]
#![allow(missing_docs)]
pub mod adaptive_selection;
pub mod benchmarking;
#[cfg(not(target_arch = "wasm32"))]
pub mod coordination;
pub mod curriculum_optimization;
#[cfg(not(target_arch = "wasm32"))]
pub mod distributed;
#[cfg(not(target_arch = "wasm32"))]
pub mod domain_specific;
pub mod error;
pub mod gpu_optimizer;
pub mod gradient_accumulation;
pub mod gradient_flow;
pub mod gradient_processing;
#[cfg(not(target_arch = "wasm32"))]
pub mod hardware_aware;
pub mod loss_landscape;
#[cfg(not(target_arch = "wasm32"))]
pub mod memory_efficient;
pub mod memory_efficient_optimizer;
pub mod metrics;
pub mod neural_integration;
#[cfg(not(target_arch = "wasm32"))]
pub mod neuromorphic;
pub mod online_learning;
pub mod optimizer_composition;
pub mod optimizer_metrics;
pub mod optimizers;
#[cfg(not(target_arch = "wasm32"))]
pub mod parallel_optimizer;
pub mod parameter_groups;
#[cfg(not(target_arch = "wasm32"))]
pub mod plugin;
#[cfg(not(target_arch = "wasm32"))]
pub mod privacy;
pub mod regularizers;
#[cfg(not(target_arch = "wasm32"))]
pub mod research;
pub mod schedulers;
pub mod second_order;
pub mod self_tuning;
pub mod simd_optimizer;
#[cfg(not(target_arch = "wasm32"))]
pub mod streaming;
pub mod training_stabilization;
pub mod unified_api;
pub mod utils;
pub mod visualization;
pub use error::{OptimError, OptimizerError, Result};
pub use optimizers::*;
pub use parameter_groups::*;
pub use regularizers::*;
pub use schedulers::*;
pub use unified_api::{OptimizerConfig, OptimizerFactory, Parameter, UnifiedOptimizer};
pub use adaptive_selection::{
AdaptiveOptimizerSelector, OptimizerStatistics, OptimizerType, PerformanceMetrics,
ProblemCharacteristics, ProblemType, SelectionNetwork, SelectionStrategy,
};
pub use curriculum_optimization::{
AdaptiveCurriculum, AdversarialAttack, AdversarialConfig, CurriculumManager, CurriculumState,
CurriculumStrategy, ImportanceWeightingStrategy,
};
#[cfg(not(target_arch = "wasm32"))]
pub use distributed::{
AveragingStrategy, CommunicationResult, CompressedGradient, CompressionStrategy,
DistributedCoordinator, GradientCompressor, ParameterAverager, ParameterServer,
};
#[cfg(not(target_arch = "wasm32"))]
pub use domain_specific::{
CrossDomainKnowledge, DomainOptimizationConfig, DomainPerformanceMetrics, DomainRecommendation,
DomainSpecificSelector, DomainStrategy, LearningRateScheduleType, OptimizationContext,
RecommendationType, RegularizationApproach, ResourceConstraints, TrainingConfiguration,
};
pub use gpu_optimizer::{GpuConfig, GpuMemoryStats, GpuOptimizer, GpuUtils};
pub use gradient_accumulation::{
AccumulationMode, GradientAccumulator as GradAccumulator, MicroBatchTrainer,
VariableAccumulator,
};
pub use gradient_processing::*;
pub use memory_efficient_optimizer::{
ChunkedOptimizer, GradientAccumulator as MemoryEfficientGradientAccumulator,
MemoryUsageEstimator,
};
pub use neural_integration::architecture_aware::{
ArchitectureAwareOptimizer, ArchitectureStrategy,
};
pub use neural_integration::forward_backward::{BackwardHook, ForwardHook, NeuralIntegration};
pub use neural_integration::{
LayerArchitecture, LayerId, OptimizationConfig, ParamId, ParameterManager, ParameterMetadata,
ParameterOptimizer, ParameterType,
};
pub use online_learning::{
ColumnGrowthStrategy, LearningRateAdaptation, LifelongOptimizer, LifelongStats,
LifelongStrategy, MemoryExample, MemoryUpdateStrategy, MirrorFunction, OnlineLearningStrategy,
OnlineOptimizer, OnlinePerformanceMetrics, SharedKnowledge, TaskGraph,
};
pub use optimizer_metrics::{
ConvergenceMetrics, GradientStatistics, MetricsCollector, MetricsReporter, OptimizerMetrics,
ParameterStatistics,
};
#[cfg(not(target_arch = "wasm32"))]
pub use parallel_optimizer::{
parallel_step, parallel_step_array1, ParallelBatchProcessor, ParallelOptimizer,
};
#[cfg(not(target_arch = "wasm32"))]
pub use plugin::core::{
create_basic_capabilities, create_plugin_info, OptimizerPluginFactory, PluginCategory,
PluginInfo,
};
#[cfg(not(target_arch = "wasm32"))]
pub use plugin::sdk::{BaseOptimizerPlugin, PluginTester};
#[cfg(not(target_arch = "wasm32"))]
pub use plugin::{
OptimizerPlugin, PluginCapabilities, PluginLoader, PluginRegistry, PluginValidationFramework,
};
#[cfg(not(target_arch = "wasm32"))]
pub use privacy::{
AccountingMethod, ClippingStats, DifferentialPrivacyConfig, DifferentiallyPrivateOptimizer,
MomentsAccountant, NoiseMechanism, PrivacyBudget, PrivacyValidation,
};
pub use second_order::{
HessianInfo, Newton, NewtonCG, SecondOrderOptimizer, LBFGS as SecondOrderLBFGS,
};
pub use self_tuning::{
OptimizerInfo, OptimizerTrait, PerformanceStats, SelfTuningConfig, SelfTuningOptimizer,
SelfTuningStatistics, TargetMetric,
};
pub use simd_optimizer::{should_use_simd, SimdOptimizer};
#[cfg(not(target_arch = "wasm32"))]
pub use streaming::{
LearningRateAdaptation as StreamingLearningRateAdaptation, StreamingConfig, StreamingDataPoint,
StreamingHealthStatus, StreamingMetrics, StreamingOptimizer,
};
pub use training_stabilization::{AveragingMethod, ModelEnsemble, PolyakAverager, WeightAverager};
pub use visualization::{
ColorScheme, ConvergenceInfo, DataSeries, MemoryStats as VisualizationMemoryStats,
OptimizationMetric, OptimizationVisualizer, OptimizerComparison, PlotType, VisualizationConfig,
};
#[cfg(feature = "metrics_integration")]
pub use metrics::*;