← all repositories

lightonai/pylate

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

834 stars Python RAG · SearchML Frameworks
pylate
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.