All modules for which code is available
- nni.algorithms.compression.pytorch.pruning.amc.amc_pruner
- nni.algorithms.compression.pytorch.pruning.auto_compress_pruner
- nni.algorithms.compression.pytorch.pruning.iterative_pruner
- nni.algorithms.compression.pytorch.pruning.lottery_ticket
- nni.algorithms.compression.pytorch.pruning.net_adapt_pruner
- nni.algorithms.compression.pytorch.pruning.one_shot_pruner
- nni.algorithms.compression.pytorch.pruning.sensitivity_pruner
- nni.algorithms.compression.pytorch.pruning.simulated_annealing_pruner
- nni.algorithms.compression.pytorch.pruning.structured_pruning_masker
- nni.algorithms.compression.pytorch.pruning.transformer_pruner
- nni.algorithms.compression.pytorch.pruning.weight_masker
- nni.algorithms.compression.pytorch.quantization.bnn_quantizer
- nni.algorithms.compression.pytorch.quantization.dorefa_quantizer
- nni.algorithms.compression.pytorch.quantization.lsq_quantizer
- nni.algorithms.compression.pytorch.quantization.native_quantizer
- nni.algorithms.compression.pytorch.quantization.observer_quantizer
- nni.algorithms.compression.pytorch.quantization.qat_quantizer
- nni.algorithms.compression.v2.pytorch.pruning.amc_pruner
- nni.algorithms.compression.v2.pytorch.pruning.auto_compress_pruner
- nni.algorithms.compression.v2.pytorch.pruning.basic_pruner
- nni.algorithms.compression.v2.pytorch.pruning.iterative_pruner
- nni.algorithms.compression.v2.pytorch.pruning.movement_pruner
- nni.algorithms.hpo.batch_tuner
- nni.algorithms.hpo.curvefitting_assessor.curvefitting_assessor
- nni.algorithms.hpo.evolution_tuner
- nni.algorithms.hpo.gp_tuner.gp_tuner
- nni.algorithms.hpo.gridsearch_tuner
- nni.algorithms.hpo.hyperband_advisor
- nni.algorithms.hpo.hyperopt_tuner
- nni.algorithms.hpo.medianstop_assessor
- nni.algorithms.hpo.metis_tuner.metis_tuner
- nni.algorithms.hpo.networkmorphism_tuner.networkmorphism_tuner
- nni.algorithms.hpo.random_tuner
- nni.algorithms.hpo.tpe_tuner
- nni.assessor
- nni.common.serializer
- nni.compression.pytorch.compressor
- nni.compression.pytorch.quantization_speedup.backend
- nni.compression.pytorch.quantization_speedup.calibrator
- nni.compression.pytorch.quantization_speedup.integrated_tensorrt
- nni.compression.pytorch.speedup.compressor
- nni.compression.pytorch.utils.counter
- nni.compression.pytorch.utils.mask_conflict
- nni.compression.pytorch.utils.sensitivity_analysis
- nni.compression.pytorch.utils.shape_dependency
- nni.nas.benchmarks.nasbench101.graph_util
- nni.nas.benchmarks.nasbench101.model
- nni.nas.benchmarks.nasbench101.query
- nni.nas.benchmarks.nasbench201.model
- nni.nas.benchmarks.nasbench201.query
- nni.nas.benchmarks.nds.model
- nni.nas.benchmarks.nds.query
- nni.retiarii.evaluator.functional
- nni.retiarii.evaluator.pytorch.cgo.evaluator
- nni.retiarii.evaluator.pytorch.lightning
- nni.retiarii.fixed
- nni.retiarii.graph
- nni.retiarii.mutator
- nni.retiarii.nn.pytorch.api
- nni.retiarii.nn.pytorch.component
- nni.retiarii.nn.pytorch.hypermodule
- nni.retiarii.oneshot.pytorch.darts
- nni.retiarii.oneshot.pytorch.enas
- nni.retiarii.oneshot.pytorch.proxyless
- nni.retiarii.oneshot.pytorch.random
- nni.retiarii.operation
- nni.retiarii.serializer
- nni.retiarii.strategy.bruteforce
- nni.retiarii.strategy.evolution
- nni.retiarii.strategy.rl
- nni.retiarii.strategy.tpe_strategy
- nni.runtime.msg_dispatcher_base
- nni.trial
- nni.tuner
- nni.utils