TorchEnsemble-Community/Ensemble-Pytorch

Health v2 · maintenance only

Adoption not part of the score

1037 stars · 95 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A unified ensemble learning framework for PyTorch that implements strategies like voting, bagging, and gradient boosting to improve deep learning model performance and robustness. It is part of the official PyTorch ecosystem and provides a scikit-learn-like API for training ensembles of neural networks.

Use cases

• improve accuracy of a pytorch neural network with ensembling
• train a voting classifier of deep learning models
• apply gradient boosting with neural network base estimators
• run bagging on pytorch models
• reduce variance of a deep learning classifier
• compare ensemble methods on image classification

When to choose

• you already use PyTorch and want ensemble methods with a familiar fit/predict API
• you need multiple ensemble strategies (voting, bagging, fusion, gradient boosting) in one library
• you want a maintained project from the official PyTorch ecosystem

When to avoid

• you work outside PyTorch (e.g., TensorFlow or JAX)
• you need ensembles of non-neural models like decision trees - use scikit-learn instead
• you need distributed or highly scaled production ensembling beyond single-machine training

Facets

library · maturity active

machine-learning deep-learning machine-learning deep-learning python ensemble-learning pytorch bagging gradient-boosting voting-classifier neural-networks

Member repositories

For agents

markdown · JSON · MCP: product_card(name="TorchEnsemble-Community/Ensemble-Pytorch")