LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
lightning_ir/__init__.py
lightning_ir/__main__.py
lightning_ir/_register_external_models.py
lightning_ir/_register_internal_models.py
lightning_ir/main.py
lightning_ir.egg-info/PKG-INFO
lightning_ir.egg-info/SOURCES.txt
lightning_ir.egg-info/dependency_links.txt
lightning_ir.egg-info/entry_points.txt
lightning_ir.egg-info/requires.txt
lightning_ir.egg-info/top_level.txt
lightning_ir/_flash/__init__.py
lightning_ir/_flash/flash_bert.py
lightning_ir/base/__init__.py
lightning_ir/base/class_factory.py
lightning_ir/base/config.py
lightning_ir/base/external_model_hub.py
lightning_ir/base/model.py
lightning_ir/base/module.py
lightning_ir/base/tokenizer.py
lightning_ir/base/validation_utils.py
lightning_ir/bi_encoder/__init__.py
lightning_ir/bi_encoder/config.py
lightning_ir/bi_encoder/model.py
lightning_ir/bi_encoder/module.py
lightning_ir/bi_encoder/tokenizer.py
lightning_ir/callbacks/__init__.py
lightning_ir/callbacks/callbacks.py
lightning_ir/cross_encoder/__init__.py
lightning_ir/cross_encoder/config.py
lightning_ir/cross_encoder/model.py
lightning_ir/cross_encoder/module.py
lightning_ir/cross_encoder/tokenizer.py
lightning_ir/data/__init__.py
lightning_ir/data/data.py
lightning_ir/data/datamodule.py
lightning_ir/data/dataset.py
lightning_ir/data/external_datasets/__init__.py
lightning_ir/data/external_datasets/colbert.py
lightning_ir/data/external_datasets/cross_architecture_knowledge_distillation.py
lightning_ir/data/external_datasets/ir_datasets_utils.py
lightning_ir/data/external_datasets/rank_distillm.py
lightning_ir/data/external_datasets/sbert.py
lightning_ir/loss/__init__.py
lightning_ir/loss/loss.py
lightning_ir/modeling_utils/__init__.py
lightning_ir/modeling_utils/mlm_head.py
lightning_ir/models/__init__.py
lightning_ir/models/col.py
lightning_ir/models/splade.py
lightning_ir/models/set_encoder/__init__.py
lightning_ir/models/set_encoder/config.py
lightning_ir/models/set_encoder/model.py
lightning_ir/models/set_encoder/tokenizer.py
lightning_ir/models/t5/__init__.py
lightning_ir/models/t5/config.py
lightning_ir/models/t5/model.py
lightning_ir/models/t5/tokenizer.py
lightning_ir/models/xtr/__init__.py
lightning_ir/models/xtr/config.py
lightning_ir/models/xtr/model.py
lightning_ir/retrieve/__init__.py
lightning_ir/retrieve/base/__init__.py
lightning_ir/retrieve/base/indexer.py
lightning_ir/retrieve/base/packed_tensor.py
lightning_ir/retrieve/base/searcher.py
lightning_ir/retrieve/faiss/__init__.py
lightning_ir/retrieve/faiss/faiss_indexer.py
lightning_ir/retrieve/faiss/faiss_searcher.py
lightning_ir/retrieve/plaid/__init__.py
lightning_ir/retrieve/plaid/plaid_indexer.py
lightning_ir/retrieve/plaid/plaid_searcher.py
lightning_ir/retrieve/plaid/residual_codec.py
lightning_ir/retrieve/pytorch/__init__.py
lightning_ir/retrieve/pytorch/dense_indexer.py
lightning_ir/retrieve/pytorch/dense_searcher.py
lightning_ir/retrieve/pytorch/sparse_indexer.py
lightning_ir/retrieve/pytorch/sparse_searcher.py
lightning_ir/retrieve/seismic/__init__.py
lightning_ir/retrieve/seismic/seismic_format.py
lightning_ir/retrieve/seismic/seismic_indexer.py
lightning_ir/retrieve/seismic/seismic_searcher.py
lightning_ir/schedulers/__init__.py
lightning_ir/schedulers/lr_schedulers.py
lightning_ir/schedulers/schedulers.py
tests/test_callbacks.py
tests/test_config.py
tests/test_data.py
tests/test_loss.py
tests/test_model.py
tests/test_schedulers.py
tests/test_tokenizer.py