Neural Architecture Search ML Frameworks

Frameworks and tools for automating neural network design through NAS techniques (DARTS, evolutionary search, graph-based methods). Does NOT include general AutoML pipeline search, hyperparameter tuning, or manual architecture guides.

There are 62 neural architecture search frameworks tracked. 1 score above 70 (verified tier). The highest-rated is pykt-team/pykt-toolkit at 70/100 with 359 stars and 3,600 monthly downloads.

Get all 62 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=neural-architecture-search&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

# Framework Score Tier
1 pykt-team/pykt-toolkit

pyKT: A Python Library to Benchmark Deep Learning based Knowledge Tracing Models

70
Verified
2 microsoft/archai

Accelerate your Neural Architecture Search (NAS) through fast, reproducible...

62
Established
3 IDEALLab/EngiBench

Benchmarks for automated engineering design

59
Established
4 AI-team-UoA/pyJedAI

An open-source library that leverages Python’s data science ecosystem to...

55
Established
5 BlackHC/batchbald_redux

Reusable BatchBALD implementation

53
Established
6 quark0/darts

Differentiable architecture search for convolutional and recurrent networks

51
Established
7 google-research/morph-net

Fast & Simple Resource-Constrained Learning of Deep Network Structure

51
Established
8 automl/neps

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

50
Established
9 ayrna/dlordinal

Open-source Python toolkit focused on deep learning with ordinal methodologies

47
Emerging
10 CDJellen/otbench

Effective Benchmarks for Optical Turbulence Modeling

45
Emerging
11 csarron/awesome-emdl

Embedded and mobile deep learning research resources

44
Emerging
12 danijar/granular

Fast dataset format and loader

44
Emerging
13 YannDubs/Neural-Process-Family

Code for the Neural Processes website and replication of 4 papers on NPs....

41
Emerging
14 jakubpeleska/redelex

ReDeLEx is a Python framework for developing and evaluating RDL models on...

41
Emerging
15 modAL-python/modAL

A modular active learning framework for Python

41
Emerging
16 GET-Foundation/get_model

A modular framework for multimodal cross-cell-type transcriptional regulation models

40
Emerging
17 ayyucedemirbas/KerasFuse

KerasFuse is a Python library that combines the power of TensorFlow and...

39
Emerging
18 llnl/dfanalyzer

Layered data-flow analysis for HPC I/O.

38
Emerging
19 aspuru-guzik-group/olympus

Olympus: a benchmarking framework for noisy optimization and experiment planning

38
Emerging
20 jinglescode/python-signal-processing

splearn: package for signal processing and machine learning with Python....

38
Emerging
21 NiuTrans/NASPapers

Paper lists of neural architecture search (NAS)

38
Emerging
22 andreazignoli/pyoxynet

The Oxynet Python package repository

36
Emerging
23 markdtw/awesome-architecture-search

A curated list of awesome architecture search resources

34
Emerging
24 anulum/scpn-fusion-core

SCPN Fusion Core — Advanced Plasma Physics & Neuromorphic Control Suite for...

34
Emerging
25 yangalan123/TemporalPointProcessPapers

Paper lists for Temporal Point Process

34
Emerging
26 HayeonLee/MetaD2A

Official PyTorch implementation of "Rapid Neural Architecture Search by...

33
Emerging
27 naotoo1/Beyond-Neural-Scaling

Implementation of Beyond Neural Scaling beating power laws for deep models...

33
Emerging
28 pietrolesci/energizer

An active learning library for Pytorch based on Lightning-Fabric.

32
Emerging
29 TimRoith/BregmanLearning

Optimizing neural networks via an inverse scale space flow.

31
Emerging
30 mercator-ocean/oceanbench

Benchmark evaluating ocean forecasting systems against reference datasets...

31
Emerging
31 automl/hierarchical_nas_construction

Official repository for "Construction of Hierarchical Neural Architecture...

30
Emerging
32 neelsomani/symbolic-circuit-distillation

Automatically extract executable programs from pruned mechanistic circuits,...

29
Experimental
33 GeminiLight/ml4co-survey

A survey on machine learning for combinatorial optimization.

28
Experimental
34 HayeonLee/HELP

Official PyTorch Implementation of HELP: Hardware-adaptive Efficient Latency...

28
Experimental
35 chenyaofo/CTNAS

[CVPR 2021] Contrastive Neural Architecture Search with Neural Architecture...

27
Experimental
36 cambridge-mlg/convcnp

Implementation of the Convolutional Conditional Neural Process

27
Experimental
37 llnl/XNAS

Explainable Neural Architecture Search (XNAS)

25
Experimental
38 T0chka/MLPermutationSolver

GPU-native framework for solving permutation sorting problems on Cayley graphs

25
Experimental
39 RylanSchaeffer/Stanford-AI-Alignment-Double-Descent-Tutorial

Code for Arxiv Double Descent Demystified: Identifying, Interpreting &...

24
Experimental
40 VITA-Group/WeakNAS

[NeurIPS 2021] “Stronger NAS with Weaker Predictors“, Junru Wu, Xiyang Dai,...

24
Experimental
41 inikishev/visualbench

Fast benchmarks for optimization algorithms with visualizations. Supports...

23
Experimental
42 VincentGranville/Point-Processes

This repository contains the material (datasets, code, videos, spreadsheets)...

23
Experimental
43 sun-umn/NCVX

NCVX: A User-Friendly and Scalable Package for Nonconvex Optimization in...

22
Experimental
44 lims2019/Arith-DAS

🚀 Optimize arithmetic circuits efficiently with differentiable architecture...

22
Experimental
45 Panopan09/pq

pq is a tool to help install and manage podman quadlets from a git...

22
Experimental
46 tonygracious/Temporal-Point-Process-TPP-

This repository contains recent background materials, current works, and...

22
Experimental
47 jwyyy/AdaFNN

[ICML 2021] Deep Learning for Functional Data Analysis with Adaptive Basis Layers

21
Experimental
48 Ren-Research/LOMAR

[ICML 2023] Learning for Edge-Weighted Online Bipartite Matching with...

21
Experimental
49 meowoodie/Neural-Spectral-Marked-Point-Processes

A novel general non-stationary point process model based on neural networks.

20
Experimental
50 jakobringler/H_ML_Castles

ML Supported Procedural Castles in SideFX Houdini using PyTorch

18
Experimental
51 NachoPeinador/Topological-Reinforcement-Operator

A neuro-inspired computational framework for modeling memory consolidation...

17
Experimental
52 FlowFrontiers/AutoFlow

Supporting page for the manuscript titled, "AutoFlow: An Autoencoder-based...

15
Experimental
53 CownowAn/DaSS

Official PyTorch implementation of "Meta-prediction Model for...

15
Experimental
54 mindfiredigital/neo-pusher

Python package designed to facilitate the seamless transfer of data from CSV...

15
Experimental
55 NzoCs/Learning-point-processes

New-LTPP is a modern, advanced framework for Temporal Point Process (TPP)...

15
Experimental
56 Lare1998/neural-architecture-search

Research implementation of efficient NAS for mobile devices.

14
Experimental
57 atomicarchitects/PriceofFreedom

[ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs...

14
Experimental
58 Pects1949/NAS-PyTorch

A PyTorch-based framework for Neural Architecture Search (NAS) to automate...

14
Experimental
59 chenyaofo/awesome-architecture-search

An up-to-date list of progress made in neural architecture search

14
Experimental
60 Ren-Research/OneProxy

[SIGMETRICS 2022] One Proxy Device Is Enough for Hardware-Aware Neural...

14
Experimental
61 jmaczan/darts-toolkit

Differentiable Architecture Search Toolkit in PyTorch Lightning

11
Experimental
62 zacharie410/RITNA

The Real-Time Information Integration for Neural Adaptation (RITNA) model is...

10
Experimental