lightonai/pylate
PyLate is a library for training and retrieving documents using ColBERT late-interaction models.

Velocity · 7d
+1.1
★ / day
Trend
→steady
star history
PyLate provides tools for fine-tuning, inference, and retrieval with ColBERT models. Built on Sentence Transformers, it supports single and multi-GPU training with contrastive loss. The library enables constructing ColBERT-style retrieval models from most pre-trained language models, streamlining document retrieval workflows commonly used in RAG pipelines.