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 61 neural architecture search frameworks tracked. 9 score above 50 (established tier). The highest-rated is pykt-team/pykt-toolkit at 69/100 with 359 stars.

Get all 61 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

69
Established
2 microsoft/archai

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

65
Established
3 google-research/morph-net

Fast & Simple Resource-Constrained Learning of Deep Network Structure

58
Established
4 AI-team-UoA/pyJedAI

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

55
Established
5 IDEALLab/EngiBench

Benchmarks for automated engineering design

55
Established
6 automl/neps

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

54
Established
7 BlackHC/batchbald_redux

Reusable BatchBALD implementation

52
Established
8 quark0/darts

Differentiable architecture search for convolutional and recurrent networks

51
Established
9 csarron/awesome-emdl

Embedded and mobile deep learning research resources

51
Established
10 CDJellen/otbench

Effective Benchmarks for Optical Turbulence Modeling

49
Emerging
11 modAL-python/modAL

A modular active learning framework for Python

48
Emerging
12 YannDubs/Neural-Process-Family

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

48
Emerging
13 jakubpeleska/redelex

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

48
Emerging
14 danijar/granular

Fast dataset format and loader

48
Emerging
15 ayrna/dlordinal

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

47
Emerging
16 GET-Foundation/get_model

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

47
Emerging
17 jinglescode/python-signal-processing

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

45
Emerging
18 NiuTrans/NASPapers

Paper lists of neural architecture search (NAS)

45
Emerging
19 aspuru-guzik-group/olympus

Olympus: a benchmarking framework for noisy optimization and experiment planning

45
Emerging
20 ayyucedemirbas/KerasFuse

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

43
Emerging
21 andreazignoli/pyoxynet

The Oxynet Python package repository

43
Emerging
22 llnl/dfanalyzer

Layered data-flow analysis for HPC I/O.

41
Emerging
23 markdtw/awesome-architecture-search

A curated list of awesome architecture search resources

41
Emerging
24 yangalan123/TemporalPointProcessPapers

Paper lists for Temporal Point Process

41
Emerging
25 HayeonLee/MetaD2A

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

40
Emerging
26 naotoo1/Beyond-Neural-Scaling

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

40
Emerging
27 pietrolesci/energizer

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

39
Emerging
28 TimRoith/BregmanLearning

Optimizing neural networks via an inverse scale space flow.

38
Emerging
29 automl/hierarchical_nas_construction

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

37
Emerging
30 mercator-ocean/oceanbench

Benchmark evaluating ocean forecasting systems against reference datasets...

35
Emerging
31 HayeonLee/HELP

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

35
Emerging
32 GeminiLight/ml4co-survey

A survey on machine learning for combinatorial optimization.

35
Emerging
33 chenyaofo/CTNAS

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

34
Emerging
34 cambridge-mlg/convcnp

Implementation of the Convolutional Conditional Neural Process

34
Emerging
35 neelsomani/symbolic-circuit-distillation

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

33
Emerging
36 anulum/scpn-fusion-core

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

33
Emerging
37 llnl/XNAS

Explainable Neural Architecture Search (XNAS)

32
Emerging
38 T0chka/MLPermutationSolver

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

31
Emerging
39 VITA-Group/WeakNAS

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

31
Emerging
40 RylanSchaeffer/Stanford-AI-Alignment-Double-Descent-Tutorial

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

31
Emerging
41 VincentGranville/Point-Processes

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

30
Emerging
42 sun-umn/NCVX

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

29
Experimental
43 inikishev/visualbench

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

29
Experimental
44 tonygracious/Temporal-Point-Process-TPP-

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

29
Experimental
45 Ren-Research/LOMAR

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

28
Experimental
46 jwyyy/AdaFNN

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

28
Experimental
47 meowoodie/Neural-Spectral-Marked-Point-Processes

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

27
Experimental
48 jakobringler/H_ML_Castles

ML Supported Procedural Castles in SideFX Houdini using PyTorch

25
Experimental
49 NachoPeinador/Topological-Reinforcement-Operator

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

23
Experimental
50 lims2019/Arith-DAS

🚀 Optimize arithmetic circuits efficiently with differentiable architecture...

22
Experimental
51 NzoCs/Learning-point-processes

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

22
Experimental
52 FlowFrontiers/AutoFlow

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

22
Experimental
53 mindfiredigital/neo-pusher

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

22
Experimental
54 CownowAn/DaSS

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

22
Experimental
55 Ren-Research/OneProxy

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

21
Experimental
56 atomicarchitects/PriceofFreedom

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

21
Experimental
57 chenyaofo/awesome-architecture-search

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

21
Experimental
58 jmaczan/darts-toolkit

Differentiable Architecture Search Toolkit in PyTorch Lightning

18
Experimental
59 zacharie410/RITNA

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

17
Experimental
60 Lare1998/neural-architecture-search

Research implementation of efficient NAS for mobile devices.

14
Experimental
61 Pects1949/NAS-PyTorch

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

14
Experimental