Hummingbird: A library for compiling trained traditional machine learning models into tensor computations

This is really a cool work out of Microsoft research called hummingbird. You can convert traditional machine learning models to tensor computations to take advantage of hardware acceleration like GPUs and TPUs. It allows users to seamlessly leverage neural network frameworks (such as PyTorch) to accelerate traditional ML models. 

It has many features and benefits as follows:

  • User can benefit from current and future optimizations implemented in neural network frameworks;
  • User can benefit from native hardware acceleration;
  • User can benefit from having a unique platform to support both traditional and neural network models;
  • User does not have to re-engineer their models.

Hummingbird is compatible with a number of tree-based classifiers and regressors. These models include scikit-learn Decision Trees and Random Forest.



Here they convert a random forest model to PyTorch (

Asif Razzaq is the CEO of Marktechpost Media Inc.. As a visionary entrepreneur and engineer, Asif is committed to harnessing the potential of Artificial Intelligence for social good. His most recent endeavor is the launch of an Artificial Intelligence Media Platform, Marktechpost, which stands out for its in-depth coverage of machine learning and deep learning news that is both technically sound and easily understandable by a wide audience. The platform boasts of over 2 million monthly views, illustrating its popularity among audiences.

🐝 Join the Fastest Growing AI Research Newsletter Read by Researchers from Google + NVIDIA + Meta + Stanford + MIT + Microsoft and many others...