All modules for which code is available
- builtins
- 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.base.pruner
- nni.algorithms.compression.v2.pytorch.base.scheduler
- 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.compression.v2.pytorch.pruning.tools.base
- nni.algorithms.compression.v2.pytorch.utils.pruning
- nni.algorithms.hpo.batch_tuner
- nni.algorithms.hpo.bohb_advisor.bohb_advisor
- nni.algorithms.hpo.curvefitting_assessor.curvefitting_assessor
- nni.algorithms.hpo.dngo_tuner
- 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.pbt_tuner
- nni.algorithms.hpo.ppo_tuner.ppo_tuner
- nni.algorithms.hpo.random_tuner
- nni.algorithms.hpo.smac_tuner.smac_tuner
- nni.algorithms.hpo.tpe_tuner
- nni.assessor
- nni.common.serializer
- nni.compression.pytorch.compressor
- 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.experiment.experiment
- 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.execution.api
- nni.retiarii.experiment.config.experiment_config
- nni.retiarii.experiment.pytorch
- nni.retiarii.fixed
- nni.retiarii.graph
- nni.retiarii.mutator
- nni.retiarii.nn.pytorch.api
- nni.retiarii.nn.pytorch.cell
- nni.retiarii.nn.pytorch.component
- nni.retiarii.nn.pytorch.hypermodule
- nni.retiarii.nn.pytorch.mutation_utils
- nni.retiarii.nn.pytorch.nasbench101
- nni.retiarii.oneshot.interface
- nni.retiarii.oneshot.pytorch.base_lightning
- nni.retiarii.oneshot.pytorch.darts
- nni.retiarii.oneshot.pytorch.dataloader
- nni.retiarii.oneshot.pytorch.enas
- nni.retiarii.oneshot.pytorch.proxyless
- nni.retiarii.oneshot.pytorch.random
- nni.retiarii.oneshot.pytorch.strategy
- nni.retiarii.oneshot.pytorch.supermodule.base
- nni.retiarii.oneshot.pytorch.supermodule.differentiable
- nni.retiarii.oneshot.pytorch.supermodule.operation
- nni.retiarii.oneshot.pytorch.supermodule.proxyless
- nni.retiarii.oneshot.pytorch.supermodule.sampling
- nni.retiarii.oneshot.pytorch.utils
- nni.retiarii.operation
- nni.retiarii.serializer
- nni.retiarii.strategy.base
- nni.retiarii.strategy.bruteforce
- nni.retiarii.strategy.evolution
- nni.retiarii.strategy.rl
- nni.retiarii.strategy.tpe_strategy
- nni.retiarii.utils
- nni.trial
- nni.tuner
- pytorch_lightning.trainer.trainer
- torch.utils.data.dataloader