#31 · Primary category: Classical Machine Learning Frameworks

pecos

approximate-nearest-neighbor-search extreme-multi-label-classification extreme-multi-label-ranking machine-learning-algorithms transformers

PECOS - Prediction for Enormous and Correlated Spaces

Project last updated:08/26/26

GitHub Stars

550

Forks

112

Contributors

32

License

Apache-2.0

Why we included this project

PECOS is built for a problem most ML libraries don't handle well: predicting from an output space that can reach hundreds of millions of candidates. It focuses on extreme multi-label ranking, where the task is to surface a small set of relevant labels from an enormous pool, and it offers two complementary approaches. X-Linear uses recursive linear models with fast C++ inference, while XR-Transformer fine-tunes pretrained transformers on multi-resolution objectives. The hierarchical label-tree design keeps both training and inference tractable at scale, which matters for teams doing product search, document tagging, or recommendation-style ranking. A bundled HNSW approximate nearest neighbor module handles sparse and dense features, so you get a practical retrieval tool alongside the ranking models.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category