Auto-PyTorch

Overview

Auto-PyTorch automates architecture search and hyperparameter optimization for PyTorch models. It uses Bayesian optimization and meta-learning to find high-performing neural network configurations without manual tuning.

Best for

Best for
PyTorch developers who want to automate hyperparameter tuning for tabular deep learning models

Use cases

• Automating hyperparameter search for custom PyTorch models
• Finding optimal neural network architectures for tabular data
• Benchmarking model performance with minimal manual intervention

Notes

Auto-PyTorch automates architecture search and hyperparameter optimization for PyTorch models. It uses Bayesian optimization and meta-learning to find high-performing neural network configurations without manual tuning.

2,534 stars on GitHub. Last updated 2024-04-09. Licensed Apache-2.0.

Use cases

• Automating hyperparameter search for custom PyTorch models
• Finding optimal neural network architectures for tabular data
• Benchmarking model performance with minimal manual intervention

Pros

• Reduces manual tuning effort with Bayesian optimization
• Integrates directly with PyTorch workflows
• Open source with active community support

Cons

• Limited to tabular data tasks, not image or text
• Search process can be computationally expensive
• Documentation and examples are sparse

Indexed from awesome-llmops and enriched against its public facts.