Chris Collins

Code AI, ML and NLP

GBDT Accelerated Ranker Framework

This framework is designed to accelerate ad and document ranking tasks using trained XGBoost models, leveraging hardware acceleration on NVIDIA CUDA GPUs, Apple Silicon Metal GPUs, and single-threaded C for portability. The project aims to improve performance by concurrently scoring user-feature vectors against large candidate document vectors i...

View on GitHub

Repository
geekychris/gbdt_accelerated_ranker_framework
Language
Python
Last updated

xgboostrankinggpucuda