ThunderGBM
A fast GPU-accelerated library for training Gradient Boosting Decision Trees (GBDT) and Random Forests.
What is ThunderGBM?
ThunderGBM is a high-performance machine learning library that trains Gradient Boosting Decision Trees (GBDT) and Random Forests using GPU acceleration. It solves the computational bottleneck of these popular tree-based ensemble methods, delivering speedups often 10x faster than other libraries for large-scale data. The library provides a scikit-learn compatible Python interface and supports classification, regression, and ranking tasks.
Data scientists and machine learning practitioners working with large datasets who need to train GBDT or Random Forest models quickly, particularly those in competitive data science (e.g., Kaggle) or real-world applications requiring fast experimentation and deployment. It is also suitable for researchers and developers seeking GPU-accelerated implementations of these algorithms.
Developers choose ThunderGBM for its significant GPU-accelerated performance gains, often 10x faster than CPU-based alternatives, making it ideal for large-scale model training. Its seamless scikit-learn compatibility allows easy integration into existing Python workflows, while its optimized parallel computing implementations efficiently handle classification, regression, and ranking problems.
Overview
ThunderGBM: Fast GBDTs and Random Forests on GPUs
Use Cases
Best For
Related Projects
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
