ENAS in PyTorch is a PyTorch implementation of Efficient Neural Architecture Search (ENAS), a method that automates the design of neural network architectures through reinforcement learning and parameter sharing. The repository demonstrates how a controller network can explore a large search space and discover high-performing architectures while dramatically reducing the computational cost traditionally associated with neural architecture search. It is primarily intended as a research and educational codebase, helping practitioners understand how ENAS works in practice and how to reproduce results on benchmark datasets. The project includes training scripts, model definitions, and search procedures that show the full workflow from architecture sampling to evaluation. Because ENAS relies on shared weights among candidate models, the implementation emphasizes efficiency and experiment reproducibility.

Features

  • PyTorch implementation of Efficient Neural Architecture Search
  • Controller-based architecture sampling workflow
  • Parameter sharing for faster search
  • Training scripts for benchmark datasets
  • Research-oriented modular code structure
  • Reproducible experiment configuration

Project Samples

Project Activity

See All Activity >

Categories

Search

License

Apache License V2.0

Follow ENAS in PyTorch

ENAS in PyTorch Web Site

Other Useful Business Software
Full-stack observability with actually useful AI | Grafana Cloud Icon
Full-stack observability with actually useful AI | Grafana Cloud

Our generous forever free tier includes the full platform, including the AI Assistant, for 3 users with 10k metrics, 50GB logs, and 50GB traces.

Built on open standards like Prometheus and OpenTelemetry, Grafana Cloud includes Kubernetes Monitoring, Application Observability, Incident Response, plus the AI-powered Grafana Assistant. Get started with our generous free tier today.
Create free account
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of ENAS in PyTorch!

Additional Project Details

Programming Language

Python

Related Categories

Python Search Software

Registered

2026-02-19