Neural Architecture Search Python Packages

nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

autokeras

AutoML library for deep learning

kubeflow-katib

Automated Machine Learning on Kubernetes

kubeflow-katib-api

Automated Machine Learning on Kubernetes

deephyper

DeepHyper: A Python Package for Massively Parallel Hyperparameter Optimization in Machine Learning

hpbandster

a distributed Hyperband implementation on Steroids

hyperactive

A unified interface for optimization algorithms and experiments

syne-tune

Large scale and asynchronous Hyperparameter and Architecture Optimization at your fingertips.

dnaty

Compress PyTorch models for edge devices — CPU-only, no GPU, no retraining. One function call.

e2eaiok

Intel® End-to-End AI Optimization Kit

e2eaiok-modeladapter

Intel® End-to-End AI Optimization Kit

betty-ml

Betty: an automatic differentiation library for generalized meta-learning and multilevel optimization

hypernets

A General Automated Machine Learning framework to simplify the development of End-to-end AutoML toolkits in specific domains.

neural-pipeline-search

Neural Pipeline Search (NePS): Helps deep learning experts find the best neural pipeline.

adanet

Fast and flexible AutoML with learning guarantees.

e2eaiok-denas

Intel® End-to-End AI Optimization Kit

e2eaiok-sda

Intel® End-to-End AI Optimization Kit

neural-dsl

A causal inference engine for deep learning training that provides structured explanations of neural network training failures. Understand why your model failed during training through semantic analysis and abductive reasoning, not raw tensor inspection.

myylearn

A General Automated Machine Learning framework to simplify the development of End-to-end AutoML toolkits in specific domains.

nni-daily

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

rfa-toolbox

A toolbox for receptive field analysis and visualizing neural network architectures

neuropt

⚡ LLM-guided ML optimization. Point it at a training script, it reads the curves and designs better models

deltatuner

Intel extension for peft with PyTorch and DENAS

xautodl

Automated deep learning algorithms implemented in PyTorch.