LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
lightning_ir/__init__.py
lightning_ir/__main__.py
lightning_ir/main.py
lightning_ir/register_external_models.py
lightning_ir/register_internal_models.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/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/bi_encoder_config.py
lightning_ir/bi_encoder/bi_encoder_model.py
lightning_ir/bi_encoder/bi_encoder_module.py
lightning_ir/bi_encoder/bi_encoder_tokenizer.py
lightning_ir/callbacks/__init__.py
lightning_ir/callbacks/callbacks.py
lightning_ir/cross_encoder/__init__.py
lightning_ir/cross_encoder/cross_encoder_config.py
lightning_ir/cross_encoder/cross_encoder_model.py
lightning_ir/cross_encoder/cross_encoder_module.py
lightning_ir/cross_encoder/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/loss/__init__.py
lightning_ir/loss/approximate.py
lightning_ir/loss/base.py
lightning_ir/loss/embedding.py
lightning_ir/loss/in_batch.py
lightning_ir/loss/listwise.py
lightning_ir/loss/neural.py
lightning_ir/loss/pairwise.py
lightning_ir/loss/regularization.py
lightning_ir/modeling_utils/__init__.py
lightning_ir/modeling_utils/batching.py
lightning_ir/modeling_utils/mlm_head.py
lightning_ir/models/__init__.py
lightning_ir/models/bi_encoders/__init__.py
lightning_ir/models/bi_encoders/coil.py
lightning_ir/models/bi_encoders/col.py
lightning_ir/models/bi_encoders/dpr.py
lightning_ir/models/bi_encoders/mvr.py
lightning_ir/models/bi_encoders/splade.py
lightning_ir/models/cross_encoders/__init__.py
lightning_ir/models/cross_encoders/mono.py
lightning_ir/models/cross_encoders/set_encoder.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