.gitignore
.pre-commit-config.yaml
LICENSE
Makefile
README.md
mkdocs.yml
pyproject.toml
test_faiss.py
.github/workflows/documentation.yaml
.github/workflows/linter.yml
.github/workflows/plaid-tests.yml
.github/workflows/publish.yml
.github/workflows/python-tests.yml
.vscode/launch.json
.vscode/settings.json
docs/.pages
docs/CNAME
docs/index.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/beir_dataset_xtr.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
examples/train/xtr_contrastive.py
examples/train/xtr_knowledge_distillation.py
examples/train/ColBERT-zero/distillation.py
examples/train/ColBERT-zero/supervised.py
examples/train/ColBERT-zero/unsupervised.py
examples/train/lateon_code/fine_tuning.py
examples/train/lateon_code/pre-training.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/scann.py
pylate/indexes/stanford_plaid.py
pylate/indexes/utils.py
pylate/indexes/voyager.py
pylate/indexes/warp.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/torch_utils.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/base.py
pylate/retrieve/colbert.py
pylate/retrieve/xtr.py
pylate/scores/__init__.py
pylate/scores/colbert.py
pylate/scores/similarity_functions.py
pylate/scores/xtr.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_colbert_scores.py
tests/test_contrastive.py
tests/test_delete_parity.py
tests/test_fast_plaid.py
tests/test_kd.py
tests/test_model.py
tests/test_model_loading.py
tests/test_plaid.py
tests/test_prefixes.py
tests/test_retriever.py
tests/test_scann.py
tests/test_warp.py
tests/test_xtr_retriever.py
tests/test_xtr_scores.py
tests/test_xtr_scoring.py