.gitignore
.pre-commit-config.yaml
LICENSE
Makefile
README.md
mkdocs.yml
pyproject.toml
test_faiss.py
.github/workflows/linter.yml
.github/workflows/plaid-tests.yml
.github/workflows/python-tests.yml
.vscode/launch.json
.vscode/settings.json
docs/.pages
docs/CNAME
docs/index.md
docs/api/.pages
docs/api/overview.md
docs/api/evaluation/.pages
docs/api/evaluation/ColBERTDistillationEvaluator.md
docs/api/evaluation/ColBERTTripletEvaluator.md
docs/api/evaluation/NanoBEIREvaluator.md
docs/api/evaluation/PyLateInformationRetrievalEvaluator.md
docs/api/evaluation/evaluate.md
docs/api/evaluation/get-beir-triples.md
docs/api/evaluation/load-beir.md
docs/api/evaluation/load-custom-dataset.md
docs/api/hf-hub/.pages
docs/api/hf-hub/PylateModelCardData.md
docs/api/indexes/.pages
docs/api/indexes/PLAID.md
docs/api/indexes/Voyager.md
docs/api/losses/.pages
docs/api/losses/CachedContrastive.md
docs/api/losses/Contrastive.md
docs/api/losses/Distillation.md
docs/api/models/.pages
docs/api/models/ColBERT.md
docs/api/models/Dense.md
docs/api/rank/.pages
docs/api/rank/rerank.md
docs/api/retrieve/.pages
docs/api/retrieve/ColBERT.md
docs/api/scores/.pages
docs/api/scores/SimilarityFunction.md
docs/api/scores/colbert-kd-scores.md
docs/api/scores/colbert-scores-pairwise.md
docs/api/scores/colbert-scores.md
docs/api/utils/.pages
docs/api/utils/ColBERTCollator.md
docs/api/utils/KDProcessing.md
docs/api/utils/all-gather-with-gradients.md
docs/api/utils/all-gather.md
docs/api/utils/convert-to-tensor.md
docs/api/utils/get-rank.md
docs/api/utils/get-world-size.md
docs/api/utils/iter-batch.md
docs/css/version-select.css
docs/documentation/.pages
docs/documentation/datasets.md
docs/documentation/evaluation.md
docs/documentation/fastapi.md
docs/documentation/retrieval.md
docs/documentation/training.md
docs/img/favicon.png
docs/img/logo.png
docs/javascripts/config.js
docs/javascripts/tablesort.js
docs/js/version-select.js
docs/models/.pages
docs/models/models.md
docs/parse/__main__.py
docs/stylesheets/extra.css
examples/evaluation/beir_dataset.py
examples/evaluation/custom_dataset.py
examples/evaluation/longembed_dataset.py
examples/train/contrastive.py
examples/train/gte_modern_colbert.py
examples/train/knowledge_distillation.py
examples/train/reason_moderncolbert.py
pylate/__init__.py
pylate/__version__.py
pylate/py.typed
pylate.egg-info/PKG-INFO
pylate.egg-info/SOURCES.txt
pylate.egg-info/dependency_links.txt
pylate.egg-info/requires.txt
pylate.egg-info/top_level.txt
pylate/evaluation/__init__.py
pylate/evaluation/beir.py
pylate/evaluation/colbert_distillation.py
pylate/evaluation/colbert_triplet.py
pylate/evaluation/custom_dataset.py
pylate/evaluation/nano_beir_evaluator.py
pylate/evaluation/pylate_information_retrieval_evaluator.py
pylate/hf_hub/__init__.py
pylate/hf_hub/model_card.py
pylate/hf_hub/model_card_template.md
pylate/indexes/__init__.py
pylate/indexes/base.py
pylate/indexes/fast_plaid.py
pylate/indexes/plaid.py
pylate/indexes/stanford_plaid.py
pylate/indexes/voyager.py
pylate/indexes/stanford_nlp/__init__.py
pylate/indexes/stanford_nlp/index_updater.py
pylate/indexes/stanford_nlp/indexer.py
pylate/indexes/stanford_nlp/parameters.py
pylate/indexes/stanford_nlp/searcher.py
pylate/indexes/stanford_nlp/data/__init__.py
pylate/indexes/stanford_nlp/data/collection.py
pylate/indexes/stanford_nlp/indexing/collection_indexer.py
pylate/indexes/stanford_nlp/indexing/index_manager.py
pylate/indexes/stanford_nlp/indexing/index_saver.py
pylate/indexes/stanford_nlp/indexing/loaders.py
pylate/indexes/stanford_nlp/indexing/utils.py
pylate/indexes/stanford_nlp/indexing/codecs/decompress_residuals.cpp
pylate/indexes/stanford_nlp/indexing/codecs/decompress_residuals.cu
pylate/indexes/stanford_nlp/indexing/codecs/packbits.cpp
pylate/indexes/stanford_nlp/indexing/codecs/packbits.cu
pylate/indexes/stanford_nlp/indexing/codecs/residual.py
pylate/indexes/stanford_nlp/indexing/codecs/residual_embeddings.py
pylate/indexes/stanford_nlp/indexing/codecs/residual_embeddings_strided.py
pylate/indexes/stanford_nlp/infra/__init__.py
pylate/indexes/stanford_nlp/infra/launcher.py
pylate/indexes/stanford_nlp/infra/run.py
pylate/indexes/stanford_nlp/infra/config/__init__.py
pylate/indexes/stanford_nlp/infra/config/base_config.py
pylate/indexes/stanford_nlp/infra/config/config.py
pylate/indexes/stanford_nlp/infra/config/core_config.py
pylate/indexes/stanford_nlp/infra/config/settings.py
pylate/indexes/stanford_nlp/modeling/colbert.py
pylate/indexes/stanford_nlp/modeling/segmented_maxsim.cpp
pylate/indexes/stanford_nlp/search/candidate_generation.py
pylate/indexes/stanford_nlp/search/decompress_residuals.cpp
pylate/indexes/stanford_nlp/search/filter_pids.cpp
pylate/indexes/stanford_nlp/search/index_loader.py
pylate/indexes/stanford_nlp/search/index_storage.py
pylate/indexes/stanford_nlp/search/segmented_lookup.cpp
pylate/indexes/stanford_nlp/search/strided_tensor.py
pylate/indexes/stanford_nlp/search/strided_tensor_core.py
pylate/indexes/stanford_nlp/utils/__init__.py
pylate/indexes/stanford_nlp/utils/amp.py
pylate/indexes/stanford_nlp/utils/coalesce.py
pylate/indexes/stanford_nlp/utils/distributed.py
pylate/indexes/stanford_nlp/utils/logging.py
pylate/indexes/stanford_nlp/utils/parser.py
pylate/indexes/stanford_nlp/utils/runs.py
pylate/indexes/stanford_nlp/utils/utils.py
pylate/losses/__init__.py
pylate/losses/cached_contrastive.py
pylate/losses/contrastive.py
pylate/losses/distillation.py
pylate/models/Dense.py
pylate/models/__init__.py
pylate/models/colbert.py
pylate/rank/__init__.py
pylate/rank/rank.py
pylate/retrieve/__init__.py
pylate/retrieve/colbert.py
pylate/scores/__init__.py
pylate/scores/scores.py
pylate/scores/similarity_functions.py
pylate/server/README.md
pylate/server/server.py
pylate/utils/__init__.py
pylate/utils/collator.py
pylate/utils/distributed.py
pylate/utils/huggingface_models.py
pylate/utils/iter_batch.py
pylate/utils/multi_process.py
pylate/utils/processing.py
pylate/utils/tensor.py
tests/test_cached_contrastive.py
tests/test_contrastive.py
tests/test_kd.py
tests/test_model.py
tests/test_model_loading.py
tests/test_plaid.py
tests/test_retriever.py