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/algos/__init__.py
archai/algos/darts/__init__.py
archai/algos/darts/bilevel_arch_trainer.py
archai/algos/darts/bilevel_optimizer.py
archai/algos/darts/bilevel_optimizer_slow.py
archai/algos/darts/darts_exp_runner.py
archai/algos/darts/darts_model_desc_builder.py
archai/algos/darts/mixed_op.py
archai/algos/didarts/__init__.py
archai/algos/didarts/didarts_arch_trainer.py
archai/algos/didarts/didarts_exp_runner.py
archai/algos/divnas/__init__.py
archai/algos/divnas/analyse_activations.py
archai/algos/divnas/divnas_cell.py
archai/algos/divnas/divnas_exp_runner.py
archai/algos/divnas/divnas_finalizers.py
archai/algos/divnas/divnas_model_desc_builder.py
archai/algos/divnas/divnas_rank_finalizer.py
archai/algos/divnas/divop.py
archai/algos/divnas/seqopt.py
archai/algos/divnas/wmr.py
archai/algos/gumbelsoftmax/__init__.py
archai/algos/gumbelsoftmax/gs_arch_trainer.py
archai/algos/gumbelsoftmax/gs_exp_runner.py
archai/algos/gumbelsoftmax/gs_finalizers.py
archai/algos/gumbelsoftmax/gs_model_desc_builder.py
archai/algos/gumbelsoftmax/gs_op.py
archai/algos/manual/__init__.py
archai/algos/manual/manual_evaluater.py
archai/algos/manual/manual_exp_runner.py
archai/algos/manual/manual_searcher.py
archai/algos/nasbench101/__init__.py
archai/algos/nasbench101/base_ops.py
archai/algos/nasbench101/config.py
archai/algos/nasbench101/graph_util.py
archai/algos/nasbench101/model.py
archai/algos/nasbench101/model_builder.py
archai/algos/nasbench101/model_matrix.py
archai/algos/nasbench101/model_metrics_pb2.py
archai/algos/nasbench101/model_spec.py
archai/algos/nasbench101/nasbench101_dataset.py
archai/algos/nasbench101/nasbench101_exp_runner.py
archai/algos/nasbench101/nasbench101_model_desc_builder.py
archai/algos/nasbench101/nasbench101_op.py
archai/algos/petridish/__init__.py
archai/algos/petridish/evaluater_petridish.py
archai/algos/petridish/petridish_exp_runner.py
archai/algos/petridish/petridish_model_desc_builder.py
archai/algos/petridish/petridish_op.py
archai/algos/petridish/petridish_utils.py
archai/algos/petridish/searcher_petridish.py
archai/algos/random/__init__.py
archai/algos/random/random_exp_runner.py
archai/algos/random/random_model_desc_builder.py
archai/algos/xnas/__init__.py
archai/algos/xnas/xnas_arch_trainer.py
archai/algos/xnas/xnas_exp_runner.py
archai/algos/xnas/xnas_model_desc_builder.py
archai/algos/xnas/xnas_op.py
archai/common/__init__.py
archai/common/apex_utils.py
archai/common/atomic_file_handler.py
archai/common/checkpoint.py
archai/common/cocob.py
archai/common/common.py
archai/common/config.py
archai/common/delimited_text.py
archai/common/lru_cache.py
archai/common/metrics.py
archai/common/ml_losses.py
archai/common/ml_perf_utils.py
archai/common/ml_utils.py
archai/common/model_summary.py
archai/common/multi_optim.py
archai/common/ordereddict_logger.py
archai/common/stopwatch.py
archai/common/tester.py
archai/common/timing.py
archai/common/trainer.py
archai/common/utils.py
archai/common/warmup_scheduler.py
archai/common/yaml_utils.py
archai/cv/__init__.py
archai/cv/augmented_search.py
archai/cv/augmented_train.py
archai/cv/datasets/__init__.py
archai/cv/datasets/aug_policies.py
archai/cv/datasets/augmentation.py
archai/cv/datasets/data.py
archai/cv/datasets/dataset_provider.py
archai/cv/datasets/distributed_stratified_sampler.py
archai/cv/datasets/limit_dataset.py
archai/cv/datasets/meta_dataset.py
archai/cv/datasets/providers/__init__.py
archai/cv/datasets/providers/aircraft_provider.py
archai/cv/datasets/providers/cifar100_provider.py
archai/cv/datasets/providers/cifar10_provider.py
archai/cv/datasets/providers/fashion_mnist_provider.py
archai/cv/datasets/providers/flower102_provider.py
archai/cv/datasets/providers/food101_provider.py
archai/cv/datasets/providers/imagenet_folder.py
archai/cv/datasets/providers/imagenet_provider.py
archai/cv/datasets/providers/mit67_provider.py
archai/cv/datasets/providers/mnist_provider.py
archai/cv/datasets/providers/person_coco_provider.py
archai/cv/datasets/providers/sport8_provider.py
archai/cv/datasets/providers/stanfordcars_provider.py
archai/cv/datasets/providers/svhn_provider.py
archai/cv/datasets/transforms/__init__.py
archai/cv/datasets/transforms/brightness.py
archai/cv/datasets/transforms/cutout_custom.py
archai/cv/datasets/transforms/lighting.py
archai/cv/models/__init__.py
archai/cv/models/alexnet.py
archai/cv/models/densenet.py
archai/cv/models/googlenet.py
archai/cv/models/inception.py
archai/cv/models/mobilenetv2.py
archai/cv/models/pyramidnet.py
archai/cv/models/resnet.py
archai/cv/models/resnet_cifar10.py
archai/cv/models/resnet_orig.py
archai/cv/models/resnet_paper.py
archai/cv/models/shakedrop.py
archai/cv/models/vgg.py
archai/cv/models/wideresnet.py
archai/cv/models/shakeshake/__init__.py
archai/cv/models/shakeshake/shake_resnet.py
archai/cv/models/shakeshake/shake_resnext.py
archai/cv/models/shakeshake/shakeshake.py
archai/cv/objectives/__init__.py
archai/cv/search_spaces/__init__.py
archai/cv/trainers/__init__.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/sucessive_halving.py
archai/discrete_search/api/__init__.py
archai/discrete_search/api/archai_model.py
archai/discrete_search/api/dataset.py
archai/discrete_search/api/objective.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/datasets/__init__.py
archai/discrete_search/datasets/lmdb_image_provider.py
archai/discrete_search/datasets/multi_lmdb_image_provider.py
archai/discrete_search/objectives/__init__.py
archai/discrete_search/objectives/functional.py
archai/discrete_search/objectives/lookup.py
archai/discrete_search/objectives/onnx_model.py
archai/discrete_search/objectives/progressive_training.py
archai/discrete_search/objectives/ray.py
archai/discrete_search/objectives/remote_azure_benchmark.py
archai/discrete_search/objectives/torch_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/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/segmentation_dag/__init__.py
archai/discrete_search/search_spaces/segmentation_dag/discrete_search_space_segmentation.py
archai/discrete_search/search_spaces/segmentation_dag/model.py
archai/discrete_search/search_spaces/segmentation_dag/ops.py
archai/discrete_search/search_spaces/segmentation_dag/search_space.py
archai/discrete_search/utils/__init__.py
archai/discrete_search/utils/evaluation.py
archai/discrete_search/utils/multi_objective.py
archai/nas/__init__.py
archai/nas/arch_module.py
archai/nas/arch_params.py
archai/nas/arch_trainer.py
archai/nas/cell.py
archai/nas/dag_edge.py
archai/nas/evaluater.py
archai/nas/exp_runner.py
archai/nas/finalizers.py
archai/nas/model.py
archai/nas/model_desc.py
archai/nas/model_desc_builder.py
archai/nas/nas_utils.py
archai/nas/operations.py
archai/nas/random_finalizers.py
archai/nas/search_combinations.py
archai/nas/searcher.py
archai/nas/vis_model_desc.py
archai/nlp/__init__.py
archai/nlp/file_utils.py
archai/nlp/logging_utils.py
archai/nlp/datasets/__init__.py
archai/nlp/datasets/hf/__init__.py
archai/nlp/datasets/hf/loaders.py
archai/nlp/datasets/hf/processors.py
archai/nlp/datasets/hf/tokenizer_utils/__init__.py
archai/nlp/datasets/hf/tokenizer_utils/bert_tokenizer.py
archai/nlp/datasets/hf/tokenizer_utils/char_tokenizer.py
archai/nlp/datasets/hf/tokenizer_utils/codegen_tokenizer.py
archai/nlp/datasets/hf/tokenizer_utils/gpt2_tokenizer.py
archai/nlp/datasets/hf/tokenizer_utils/pre_trained_tokenizer.py
archai/nlp/datasets/hf/tokenizer_utils/token_config.py
archai/nlp/datasets/hf/tokenizer_utils/tokenizer_base.py
archai/nlp/datasets/hf/tokenizer_utils/transfo_xl_tokenizer.py
archai/nlp/datasets/nvidia/__init__.py
archai/nlp/datasets/nvidia/corpus.py
archai/nlp/datasets/nvidia/corpus_utils.py
archai/nlp/datasets/nvidia/distributed_utils.py
archai/nlp/datasets/nvidia/lm_iterators.py
archai/nlp/datasets/nvidia/tokenizer_utils/__init__.py
archai/nlp/datasets/nvidia/tokenizer_utils/bbpe_vocab.py
archai/nlp/datasets/nvidia/tokenizer_utils/gpt2_vocab.py
archai/nlp/datasets/nvidia/tokenizer_utils/token_config.py
archai/nlp/datasets/nvidia/tokenizer_utils/vocab_base.py
archai/nlp/datasets/nvidia/tokenizer_utils/word_vocab.py
archai/nlp/eval/__init__.py
archai/nlp/eval/eval_utils.py
archai/nlp/eval/onnx_evaluator.py
archai/nlp/eval/torch_evaluator.py
archai/nlp/eval/harness/__init__.py
archai/nlp/eval/harness/harness_eval.py
archai/nlp/eval/harness/harness_model.py
archai/nlp/eval/harness/harness_task.py
archai/nlp/eval/harness/harness_utils.py
archai/nlp/eval/harness/tasks/__init__.py
archai/nlp/eval/harness/tasks/arc.py
archai/nlp/eval/harness/tasks/glue.py
archai/nlp/eval/harness/tasks/hella_swag.py
archai/nlp/eval/harness/tasks/human_eval.py
archai/nlp/eval/harness/tasks/lambada.py
archai/nlp/eval/harness/tasks/open_book_qa.py
archai/nlp/eval/harness/tasks/piqa.py
archai/nlp/eval/harness/tasks/story_cloze.py
archai/nlp/eval/harness/tasks/super_glue.py
archai/nlp/eval/harness/tasks/wino_grande.py
archai/nlp/eval/profiler/__init__.py
archai/nlp/eval/profiler/profiler_eval.py
archai/nlp/eval/profiler/profiler_model.py
archai/nlp/eval/profiler/profiler_utils.py
archai/nlp/eval/text_predict/__init__.py
archai/nlp/eval/text_predict/text_predict_eval.py
archai/nlp/eval/text_predict/text_predict_model.py
archai/nlp/eval/text_predict/text_predict_prediction.py
archai/nlp/eval/text_predict/text_predict_predictor.py
archai/nlp/eval/text_predict/text_predict_tokenizer.py
archai/nlp/eval/text_predict/text_predict_utils.py
archai/nlp/objectives/__init__.py
archai/nlp/objectives/parameters.py
archai/nlp/objectives/transformer_flex_latency.py
archai/nlp/objectives/transformer_flex_memory.py
archai/nlp/onnx/__init__.py
archai/nlp/onnx/export.py
archai/nlp/onnx/export_utils.py
archai/nlp/onnx/onnx_forward.py
archai/nlp/onnx/onnx_loader.py
archai/nlp/onnx/optimization.py
archai/nlp/onnx/config_utils/__init__.py
archai/nlp/onnx/config_utils/gpt2_onnx_config.py
archai/nlp/onnx/config_utils/onnx_config_base.py
archai/nlp/onnx/optimization_utils/__init__.py
archai/nlp/onnx/optimization_utils/fusion_options.py
archai/nlp/onnx/optimization_utils/transfo_xl_onnx_model.py
archai/nlp/quantization/__init__.py
archai/nlp/quantization/mixed_qat.py
archai/nlp/quantization/modules.py
archai/nlp/quantization/observers.py
archai/nlp/quantization/ptq.py
archai/nlp/quantization/qat.py
archai/nlp/quantization/quantization_utils.py
archai/nlp/quantization/quantizers.py
archai/nlp/search_spaces/__init__.py
archai/nlp/search_spaces/transformer_flex/__init__.py
archai/nlp/search_spaces/transformer_flex/search_space.py
archai/nlp/search_spaces/transformer_flex/models/__init__.py
archai/nlp/search_spaces/transformer_flex/models/gpt2_flex/__init__.py
archai/nlp/search_spaces/transformer_flex/models/gpt2_flex/configuration_gpt2_flex.py
archai/nlp/search_spaces/transformer_flex/models/gpt2_flex/modeling_gpt2_flex.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/__init__.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/configuration_mem_transformer.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/modeling_mem_transformer.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/utils/__init__.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/utils/adaptive_embedding.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/utils/depth_wise_convolution.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/utils/position_wise_ff.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/utils/positional_embedding.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/utils/projected_adaptive_log_softmax.py
archai/nlp/search_spaces/transformer_flex/models/mem_transformer/utils/rel_partial_learnable_decoder.py
archai/nlp/trainers/__init__.py
archai/nlp/trainers/hf/__init__.py
archai/nlp/trainers/hf/callbacks.py
archai/nlp/trainers/hf/trainer.py
archai/nlp/trainers/hf/training_args.py
archai/nlp/trainers/nvidia/__init__.py
archai/nlp/trainers/nvidia/cyclic_cosine_scheduler.py
archai/nlp/trainers/nvidia/optimizers.py
archai/nlp/trainers/nvidia/trainer.py
archai/nlp/trainers/nvidia/training_args.py