AUTHORS.md
LICENSE
NOTICE.md
README.md
pyproject.toml
setup.cfg
setup.py
archai/__init__.py
archai.egg-info/PKG-INFO
archai.egg-info/SOURCES.txt
archai.egg-info/dependency_links.txt
archai.egg-info/requires.txt
archai.egg-info/top_level.txt
archai/api/__init__.py
archai/api/dataset_provider.py
archai/api/trainer_base.py
archai/common/__init__.py
archai/common/apex_utils.py
archai/common/atomic_file_handler.py
archai/common/azureml_helper.py
archai/common/common.py
archai/common/config.py
archai/common/delimited_text.py
archai/common/deprecation_utils.py
archai/common/distributed_utils.py
archai/common/file_utils.py
archai/common/ml_perf_utils.py
archai/common/ml_utils.py
archai/common/model_summary.py
archai/common/monitor.py
archai/common/notebook_helper.py
archai/common/ordered_dict_logger.py
archai/common/ordered_dict_logger_utils.py
archai/common/stopwatch.py
archai/common/store.py
archai/common/timing.py
archai/common/utils.py
archai/common/yaml_utils.py
archai/datasets/__init__.py
archai/datasets/cv/__init__.py
archai/datasets/cv/aircraft_dataset_provider.py
archai/datasets/cv/caltech_dataset_provider.py
archai/datasets/cv/cifar_dataset_provider.py
archai/datasets/cv/cityscapes_dataset_provider.py
archai/datasets/cv/coco_dataset_provider.py
archai/datasets/cv/face_synthetics.py
archai/datasets/cv/flowers102_dataset_provider.py
archai/datasets/cv/food101_dataset_provider.py
archai/datasets/cv/image_folder_dataset_provider.py
archai/datasets/cv/imagenet_dataset_provider.py
archai/datasets/cv/mnist_dataset_provider.py
archai/datasets/cv/stanford_cars_dataset_provider.py
archai/datasets/cv/svhn_dataset_provider.py
archai/datasets/cv/tensorpack_lmdb_dataset_provider.py
archai/datasets/cv/tensorpack_lmdb_dataset_provider_utils.py
archai/datasets/cv/usps_dataset_provider.py
archai/datasets/cv/transforms/__init__.py
archai/datasets/cv/transforms/brightness.py
archai/datasets/cv/transforms/custom_cutout.py
archai/datasets/cv/transforms/lighting.py
archai/datasets/nlp/__init__.py
archai/datasets/nlp/fast_hf_dataset_provider.py
archai/datasets/nlp/fast_hf_dataset_provider_utils.py
archai/datasets/nlp/hf_dataset_provider.py
archai/datasets/nlp/hf_dataset_provider_utils.py
archai/datasets/nlp/nvidia_data_loader_utils.py
archai/datasets/nlp/nvidia_dataset_provider.py
archai/datasets/nlp/nvidia_dataset_provider_utils.py
archai/datasets/nlp/tokenizer_utils/__init__.py
archai/datasets/nlp/tokenizer_utils/bbpe_tokenizer.py
archai/datasets/nlp/tokenizer_utils/gpt2_tokenizer.py
archai/datasets/nlp/tokenizer_utils/token_config.py
archai/datasets/nlp/tokenizer_utils/tokenizer_base.py
archai/datasets/nlp/tokenizer_utils/word_tokenizer.py
archai/discrete_search/__init__.py
archai/discrete_search/algos/__init__.py
archai/discrete_search/algos/bananas.py
archai/discrete_search/algos/evolution_pareto.py
archai/discrete_search/algos/local_search.py
archai/discrete_search/algos/random_search.py
archai/discrete_search/algos/regularized_evolution.py
archai/discrete_search/algos/successive_halving.py
archai/discrete_search/api/__init__.py
archai/discrete_search/api/archai_model.py
archai/discrete_search/api/model_evaluator.py
archai/discrete_search/api/predictor.py
archai/discrete_search/api/search_objectives.py
archai/discrete_search/api/search_results.py
archai/discrete_search/api/search_space.py
archai/discrete_search/api/searcher.py
archai/discrete_search/evaluators/__init__.py
archai/discrete_search/evaluators/functional.py
archai/discrete_search/evaluators/onnx_model.py
archai/discrete_search/evaluators/progressive_training.py
archai/discrete_search/evaluators/pt_profiler.py
archai/discrete_search/evaluators/ray.py
archai/discrete_search/evaluators/remote_azure_benchmark.py
archai/discrete_search/evaluators/benchmark/__init__.py
archai/discrete_search/evaluators/benchmark/natsbench_tss.py
archai/discrete_search/evaluators/nlp/__init__.py
archai/discrete_search/evaluators/nlp/parameters.py
archai/discrete_search/evaluators/nlp/transformer_flex_latency.py
archai/discrete_search/evaluators/nlp/transformer_flex_memory.py
archai/discrete_search/evaluators/pt_profiler_utils/__init__.py
archai/discrete_search/evaluators/pt_profiler_utils/pt_profiler_eval.py
archai/discrete_search/evaluators/pt_profiler_utils/pt_profiler_hooks.py
archai/discrete_search/evaluators/pt_profiler_utils/pt_profiler_model.py
archai/discrete_search/predictors/__init__.py
archai/discrete_search/predictors/dnn_ensemble.py
archai/discrete_search/search_spaces/__init__.py
archai/discrete_search/search_spaces/benchmark/__init__.py
archai/discrete_search/search_spaces/benchmark/natsbench_tss.py
archai/discrete_search/search_spaces/config/__init__.py
archai/discrete_search/search_spaces/config/arch_config.py
archai/discrete_search/search_spaces/config/arch_param_tree.py
archai/discrete_search/search_spaces/config/discrete_choice.py
archai/discrete_search/search_spaces/config/helpers.py
archai/discrete_search/search_spaces/config/search_space.py
archai/discrete_search/search_spaces/config/utils.py
archai/discrete_search/search_spaces/cv/__init__.py
archai/discrete_search/search_spaces/cv/segmentation_dag/__init__.py
archai/discrete_search/search_spaces/cv/segmentation_dag/model.py
archai/discrete_search/search_spaces/cv/segmentation_dag/ops.py
archai/discrete_search/search_spaces/cv/segmentation_dag/search_space.py
archai/discrete_search/search_spaces/nlp/__init__.py
archai/discrete_search/search_spaces/nlp/tfpp/__init__.py
archai/discrete_search/search_spaces/nlp/tfpp/mixed_attention.py
archai/discrete_search/search_spaces/nlp/tfpp/mixed_op.py
archai/discrete_search/search_spaces/nlp/tfpp/model.py
archai/discrete_search/search_spaces/nlp/tfpp/modeling_codegen_flash.py
archai/discrete_search/search_spaces/nlp/tfpp/search_space.py
archai/discrete_search/search_spaces/nlp/tfpp/utils.py
archai/discrete_search/search_spaces/nlp/tfpp/backbones/__init__.py
archai/discrete_search/search_spaces/nlp/tfpp/backbones/codegen/__init__.py
archai/discrete_search/search_spaces/nlp/tfpp/backbones/codegen/block.py
archai/discrete_search/search_spaces/nlp/tfpp/backbones/codegen/model.py
archai/discrete_search/search_spaces/nlp/tfpp/backbones/gpt2/__init__.py
archai/discrete_search/search_spaces/nlp/tfpp/backbones/gpt2/block.py
archai/discrete_search/search_spaces/nlp/tfpp/backbones/gpt2/model.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/__init__.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/causal_self_attn.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/fftconv_.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/local_attention.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/lsh_attn.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/mha.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/sep_conv1d.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/sgconv.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/sgconv3.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/lsh_utils/__init__.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/lsh_utils/modeling_reformer.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/ssm_utils/__init__.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/ssm_utils/dplr.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/ssm_utils/hippo.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/ssm_utils/ss_kernel.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/ssm_utils/ss_kernel_diag.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/ssm_utils/ss_kernel_shift.py
archai/discrete_search/search_spaces/nlp/tfpp/ops/ssm_utils/utils.py
archai/discrete_search/search_spaces/nlp/transformer_flex/__init__.py
archai/discrete_search/search_spaces/nlp/transformer_flex/search_space.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/__init__.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/configuration_gpt2_flex.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/configuration_mem_transformer.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/modeling_gpt2_flex.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/modeling_mem_transformer.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/mem_transformer_utils/__init__.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/mem_transformer_utils/adaptive_embedding.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/mem_transformer_utils/depth_wise_convolution.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/mem_transformer_utils/position_wise_ff.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/mem_transformer_utils/positional_embedding.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/mem_transformer_utils/projected_adaptive_log_softmax.py
archai/discrete_search/search_spaces/nlp/transformer_flex/models/mem_transformer_utils/rel_partial_learnable_decoder.py
archai/discrete_search/utils/__init__.py
archai/discrete_search/utils/multi_objective.py
archai/onnx/__init__.py
archai/onnx/export.py
archai/onnx/export_utils.py
archai/onnx/onnx_forward.py
archai/onnx/onnx_loader.py
archai/onnx/optimization.py
archai/onnx/config_utils/__init__.py
archai/onnx/config_utils/codegen_onnx_config.py
archai/onnx/config_utils/gpt2_onnx_config.py
archai/onnx/config_utils/onnx_config_base.py
archai/onnx/optimization_utils/__init__.py
archai/onnx/optimization_utils/fusion_options.py
archai/onnx/optimization_utils/transfo_xl_onnx_model.py
archai/quantization/__init__.py
archai/quantization/mixed_qat.py
archai/quantization/modules.py
archai/quantization/observers.py
archai/quantization/ptq.py
archai/quantization/qat.py
archai/quantization/quantization_utils.py
archai/quantization/quantizers.py
archai/quantization/nlp/__init__.py
archai/quantization/nlp/modules.py
archai/supergraph/__init__.py
archai/supergraph/algos/__init__.py
archai/supergraph/algos/darts/__init__.py
archai/supergraph/algos/darts/bilevel_arch_trainer.py
archai/supergraph/algos/darts/bilevel_optimizer.py
archai/supergraph/algos/darts/bilevel_optimizer_slow.py
archai/supergraph/algos/darts/darts_exp_runner.py
archai/supergraph/algos/darts/darts_model_desc_builder.py
archai/supergraph/algos/darts/mixed_op.py
archai/supergraph/algos/didarts/__init__.py
archai/supergraph/algos/didarts/didarts_arch_trainer.py
archai/supergraph/algos/didarts/didarts_exp_runner.py
archai/supergraph/algos/divnas/__init__.py
archai/supergraph/algos/divnas/analyse_activations.py
archai/supergraph/algos/divnas/divnas_cell.py
archai/supergraph/algos/divnas/divnas_exp_runner.py
archai/supergraph/algos/divnas/divnas_finalizers.py
archai/supergraph/algos/divnas/divnas_model_desc_builder.py
archai/supergraph/algos/divnas/divnas_rank_finalizer.py
archai/supergraph/algos/divnas/divop.py
archai/supergraph/algos/divnas/seqopt.py
archai/supergraph/algos/divnas/wmr.py
archai/supergraph/algos/gumbelsoftmax/__init__.py
archai/supergraph/algos/gumbelsoftmax/gs_arch_trainer.py
archai/supergraph/algos/gumbelsoftmax/gs_exp_runner.py
archai/supergraph/algos/gumbelsoftmax/gs_finalizers.py
archai/supergraph/algos/gumbelsoftmax/gs_model_desc_builder.py
archai/supergraph/algos/gumbelsoftmax/gs_op.py
archai/supergraph/algos/manual/__init__.py
archai/supergraph/algos/manual/manual_evaluater.py
archai/supergraph/algos/manual/manual_exp_runner.py
archai/supergraph/algos/manual/manual_searcher.py
archai/supergraph/algos/nasbench101/__init__.py
archai/supergraph/algos/nasbench101/base_ops.py
archai/supergraph/algos/nasbench101/config.py
archai/supergraph/algos/nasbench101/graph_util.py
archai/supergraph/algos/nasbench101/model.py
archai/supergraph/algos/nasbench101/model_builder.py
archai/supergraph/algos/nasbench101/model_matrix.py
archai/supergraph/algos/nasbench101/model_metrics_pb2.py
archai/supergraph/algos/nasbench101/model_spec.py
archai/supergraph/algos/nasbench101/nasbench101_dataset.py
archai/supergraph/algos/nasbench101/nasbench101_exp_runner.py
archai/supergraph/algos/nasbench101/nasbench101_model_desc_builder.py
archai/supergraph/algos/nasbench101/nasbench101_op.py
archai/supergraph/algos/petridish/__init__.py
archai/supergraph/algos/petridish/evaluater_petridish.py
archai/supergraph/algos/petridish/petridish_exp_runner.py
archai/supergraph/algos/petridish/petridish_model_desc_builder.py
archai/supergraph/algos/petridish/petridish_op.py
archai/supergraph/algos/petridish/petridish_utils.py
archai/supergraph/algos/petridish/searcher_petridish.py
archai/supergraph/algos/random/__init__.py
archai/supergraph/algos/random/random_exp_runner.py
archai/supergraph/algos/random/random_model_desc_builder.py
archai/supergraph/algos/xnas/__init__.py
archai/supergraph/algos/xnas/xnas_arch_trainer.py
archai/supergraph/algos/xnas/xnas_exp_runner.py
archai/supergraph/algos/xnas/xnas_model_desc_builder.py
archai/supergraph/algos/xnas/xnas_op.py
archai/supergraph/datasets/__init__.py
archai/supergraph/datasets/aug_policies.py
archai/supergraph/datasets/augmentation.py
archai/supergraph/datasets/data.py
archai/supergraph/datasets/dataset_provider.py
archai/supergraph/datasets/distributed_stratified_sampler.py
archai/supergraph/datasets/limit_dataset.py
archai/supergraph/datasets/meta_dataset.py
archai/supergraph/datasets/providers/__init__.py
archai/supergraph/datasets/providers/aircraft_provider.py
archai/supergraph/datasets/providers/cifar100_provider.py
archai/supergraph/datasets/providers/cifar10_provider.py
archai/supergraph/datasets/providers/fashion_mnist_provider.py
archai/supergraph/datasets/providers/flower102_provider.py
archai/supergraph/datasets/providers/food101_provider.py
archai/supergraph/datasets/providers/imagenet_folder.py
archai/supergraph/datasets/providers/imagenet_provider.py
archai/supergraph/datasets/providers/mit67_provider.py
archai/supergraph/datasets/providers/mnist_provider.py
archai/supergraph/datasets/providers/person_coco_provider.py
archai/supergraph/datasets/providers/sport8_provider.py
archai/supergraph/datasets/providers/stanfordcars_provider.py
archai/supergraph/datasets/providers/svhn_provider.py
archai/supergraph/models/__init__.py
archai/supergraph/models/alexnet.py
archai/supergraph/models/densenet.py
archai/supergraph/models/googlenet.py
archai/supergraph/models/inception.py
archai/supergraph/models/mobilenetv2.py
archai/supergraph/models/pyramidnet.py
archai/supergraph/models/resnet.py
archai/supergraph/models/resnet_orig.py
archai/supergraph/models/resnet_paper.py
archai/supergraph/models/shakedrop.py
archai/supergraph/models/vgg.py
archai/supergraph/models/wideresnet.py
archai/supergraph/models/shakeshake/__init__.py
archai/supergraph/models/shakeshake/shake_resnet.py
archai/supergraph/models/shakeshake/shake_resnext.py
archai/supergraph/models/shakeshake/shakeshake.py
archai/supergraph/nas/__init__.py
archai/supergraph/nas/arch_module.py
archai/supergraph/nas/arch_params.py
archai/supergraph/nas/arch_trainer.py
archai/supergraph/nas/cell.py
archai/supergraph/nas/dag_edge.py
archai/supergraph/nas/evaluater.py
archai/supergraph/nas/exp_runner.py
archai/supergraph/nas/finalizers.py
archai/supergraph/nas/model.py
archai/supergraph/nas/model_desc.py
archai/supergraph/nas/model_desc_builder.py
archai/supergraph/nas/nas_utils.py
archai/supergraph/nas/operations.py
archai/supergraph/nas/random_finalizers.py
archai/supergraph/nas/search_combinations.py
archai/supergraph/nas/searcher.py
archai/supergraph/nas/vis_model_desc.py
archai/supergraph/utils/__init__.py
archai/supergraph/utils/augmented_searcher.py
archai/supergraph/utils/augmented_trainer.py
archai/supergraph/utils/checkpoint.py
archai/supergraph/utils/heatmap.py
archai/supergraph/utils/metrics.py
archai/supergraph/utils/multi_optim.py
archai/supergraph/utils/tester.py
archai/supergraph/utils/trainer.py
archai/trainers/__init__.py
archai/trainers/coin_betting_optimizer.py
archai/trainers/cyclic_cosine_scheduler.py
archai/trainers/gradual_warmup_scheduler.py
archai/trainers/lamb_optimizer.py
archai/trainers/losses.py
archai/trainers/cv/__init__.py
archai/trainers/cv/pl_trainer.py
archai/trainers/nlp/__init__.py
archai/trainers/nlp/ds_trainer.py
archai/trainers/nlp/ds_training_args.py
archai/trainers/nlp/hf_callbacks.py
archai/trainers/nlp/hf_trainer.py
archai/trainers/nlp/hf_training_args.py
archai/trainers/nlp/nvidia_trainer.py
archai/trainers/nlp/nvidia_training_args.py