Index

_

__init__() (lightning_ir.base.class_factory.LightningIRClassFactory method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory method) (lightning_ir.base.class_factory.LightningIRModelClassFactory method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory method) (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.base.model.BatchEncodingWrapper method) (lightning_ir.base.model.LightningIRModel method) (lightning_ir.base.model.LightningIROutput method) (lightning_ir.base.module.LightningIRModule method) (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.bi_encoder_config.BiEncoderConfig method) (lightning_ir.bi_encoder.bi_encoder_config.MultiVectorBiEncoderConfig method) (lightning_ir.bi_encoder.bi_encoder_config.SingleVectorBiEncoderConfig method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.cross_encoder_config.CrossEncoderConfig method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderModel method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_tokenizer.CrossEncoderTokenizer method) (lightning_ir.data.data.DocSample method) (lightning_ir.data.data.IndexBatch method) (lightning_ir.data.data.QuerySample method) (lightning_ir.data.data.RankBatch method) (lightning_ir.data.data.RankSample method) (lightning_ir.data.data.SearchBatch method) (lightning_ir.data.data.TrainBatch method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.data.dataset.DocDataset method) (lightning_ir.data.dataset.IRDataset method) (lightning_ir.data.dataset.QueryDataset method) (lightning_ir.data.dataset.RunDataset method) (lightning_ir.data.dataset.Sampler method) (lightning_ir.data.dataset.TupleDataset method) (lightning_ir.data.external_datasets.ir_datasets_utils.ParquetScoredDocs method) (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple method) (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuples method) (lightning_ir.loss.loss.ApproxLossFunction method) (lightning_ir.loss.loss.ApproxMRR method) (lightning_ir.loss.loss.ApproxNDCG method) (lightning_ir.loss.loss.ApproxRankMSE method) (lightning_ir.loss.loss.ConstantMarginMSE method) (lightning_ir.loss.loss.EmbeddingLossFunction method) (lightning_ir.loss.loss.FLOPSRegularization method) (lightning_ir.loss.loss.InBatchCrossEntropy method) (lightning_ir.loss.loss.InBatchLossFunction method) (lightning_ir.loss.loss.InfoNCE method) (lightning_ir.loss.loss.KLDivergence method) (lightning_ir.loss.loss.L1Regularization method) (lightning_ir.loss.loss.L2Regularization method) (lightning_ir.loss.loss.ListwiseLossFunction method) (lightning_ir.loss.loss.LossFunction method) (lightning_ir.loss.loss.MarginMSE method) (lightning_ir.loss.loss.NeuralLossFunction method) (lightning_ir.loss.loss.PairwiseLossFunction method) (lightning_ir.loss.loss.PearsonCorrelation method) (lightning_ir.loss.loss.RankNet method) (lightning_ir.loss.loss.RegularizationLossFunction method) (lightning_ir.loss.loss.ScoreBasedInBatchCrossEntropy method) (lightning_ir.loss.loss.ScoreBasedInBatchLossFunction method) (lightning_ir.loss.loss.ScoringLossFunction method) (lightning_ir.loss.loss.SupervisedMarginMSE method) (lightning_ir.main.LightningIRCLI method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.main.LightningIRTrainer method) (lightning_ir.main.LightningIRWandbLogger method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.col.ColTokenizer method) (lightning_ir.models.dpr.DprConfig method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.set_encoder.SetEncoderConfig method) (lightning_ir.models.set_encoder.SetEncoderModel method) (lightning_ir.models.set_encoder.SetEncoderTokenizer method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.t5_cross_encoder.ScaleLinear method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderConfig method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderModel method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderTokenizer method) (lightning_ir.retrieve.base.indexer.IndexConfig method) (lightning_ir.retrieve.base.indexer.Indexer method) (lightning_ir.retrieve.base.packed_tensor.PackedTensor method) (lightning_ir.retrieve.base.searcher.ApproximateSearchConfig method) (lightning_ir.retrieve.base.searcher.ApproximateSearcher method) (lightning_ir.retrieve.base.searcher.ExactSearchConfig method) (lightning_ir.retrieve.base.searcher.ExactSearcher method) (lightning_ir.retrieve.base.searcher.SearchConfig method) (lightning_ir.retrieve.base.searcher.Searcher method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissFlatIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissFlatIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFPQIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFPQIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexer method) (lightning_ir.retrieve.faiss.faiss_searcher.FaissSearchConfig method) (lightning_ir.retrieve.faiss.faiss_searcher.FaissSearcher method) (lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexConfig method) (lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexer method) (lightning_ir.retrieve.plaid.plaid_searcher.PlaidSearchConfig method) (lightning_ir.retrieve.plaid.plaid_searcher.PlaidSearcher method) (lightning_ir.retrieve.plaid.residual_codec.ResidualCodec method) (lightning_ir.retrieve.pytorch.dense_indexer.TorchDenseIndexConfig method) (lightning_ir.retrieve.pytorch.dense_indexer.TorchDenseIndexer method) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseIndex method) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseSearchConfig method) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseSearcher method) (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexConfig method) (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexer method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseIndex method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseSearchConfig method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseSearcher method) (lightning_ir.retrieve.seismic.seismic_format.SeismicFormatConverter method) (lightning_ir.retrieve.seismic.seismic_indexer.SeismicIndexConfig method) (lightning_ir.retrieve.seismic.seismic_indexer.SeismicIndexer method) (lightning_ir.retrieve.seismic.seismic_searcher.SeismicSearchConfig method) (lightning_ir.retrieve.seismic.seismic_searcher.SeismicSearcher method) (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) (lightning_ir.schedulers.schedulers.ConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.ConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) (lightning_ir.schedulers.schedulers.LambdaWarmupScheduler method) (lightning_ir.schedulers.schedulers.LinearSchedulerWithLinearWarmup method)

A

add_arguments_to_parser() (lightning_ir.main.LightningIRCLI method) after_save_checkpoint() (lightning_ir.main.LightningIRWandbLogger method) aggregate_similarity() (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.models.col.ColModel method) ALLOW_SUB_BATCHING (lightning_ir.base.model.LightningIRModel attribute) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel attribute) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel attribute) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel attribute) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderModel attribute) (lightning_ir.models.col.ColModel attribute) (lightning_ir.models.dpr.DprModel attribute) (lightning_ir.models.set_encoder.SetEncoderModel attribute) (lightning_ir.models.splade.SpladeModel attribute) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderModel attribute)

ApproximateSearchConfig (class in lightning_ir.retrieve.base.searcher) ApproximateSearcher (class in lightning_ir.retrieve.base.searcher) ApproxLossFunction (class in lightning_ir.loss.loss) ApproxMRR (class in lightning_ir.loss.loss) ApproxNDCG (class in lightning_ir.loss.loss) ApproxRankMSE (class in lightning_ir.loss.loss)

B

backbone_model_type (lightning_ir.base.config.LightningIRConfig attribute) (lightning_ir.bi_encoder.bi_encoder_config.BiEncoderConfig attribute) (lightning_ir.bi_encoder.bi_encoder_config.MultiVectorBiEncoderConfig attribute) (lightning_ir.bi_encoder.bi_encoder_config.SingleVectorBiEncoderConfig attribute) (lightning_ir.cross_encoder.cross_encoder_config.CrossEncoderConfig attribute) (lightning_ir.models.col.ColConfig attribute) (lightning_ir.models.dpr.DprConfig attribute) (lightning_ir.models.set_encoder.SetEncoderConfig attribute) (lightning_ir.models.splade.SpladeConfig attribute) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderConfig attribute)

batch_encoding_wrapper() (in module lightning_ir.base.model) BatchEncodingWrapper (class in lightning_ir.base.model) BiEncoderConfig (class in lightning_ir.bi_encoder.bi_encoder_config) BiEncoderEmbedding (class in lightning_ir.bi_encoder.bi_encoder_model) BiEncoderModel (class in lightning_ir.bi_encoder.bi_encoder_model) BiEncoderModule (class in lightning_ir.bi_encoder.bi_encoder_module) BiEncoderOutput (class in lightning_ir.bi_encoder.bi_encoder_model) BiEncoderTokenizer (class in lightning_ir.bi_encoder.bi_encoder_tokenizer)

C

cat_arange() (in module lightning_ir.retrieve.base.searcher) cc_lir_model_type (lightning_ir.base.class_factory.LightningIRClassFactory property) (lightning_ir.base.class_factory.LightningIRConfigClassFactory property) (lightning_ir.base.class_factory.LightningIRModelClassFactory property) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory property) CHECKPOINT_MAPPING (in module lightning_ir.base.external_model_hub) clear() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) ColConfig (class in lightning_ir.models.col) ColModel (class in lightning_ir.models.col) ColTokenizer (class in lightning_ir.models.col) compute_similarity() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.splade.SpladeModel method) config_class (lightning_ir.base.model.LightningIRModel attribute) (lightning_ir.base.tokenizer.LightningIRTokenizer attribute) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel attribute) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel attribute) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel attribute) (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer attribute) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderModel attribute) (lightning_ir.cross_encoder.cross_encoder_tokenizer.CrossEncoderTokenizer attribute) (lightning_ir.models.col.ColModel attribute) (lightning_ir.models.col.ColTokenizer attribute) (lightning_ir.models.dpr.DprModel attribute) (lightning_ir.models.set_encoder.SetEncoderModel attribute) (lightning_ir.models.set_encoder.SetEncoderTokenizer attribute) (lightning_ir.models.splade.SpladeModel attribute) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderModel attribute) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderTokenizer attribute)

configure_optimizers() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) (lightning_ir.main.LightningIRCLI static method) ConstantLRSchedulerWithLinearWarmup (class in lightning_ir.schedulers.lr_schedulers) ConstantMarginMSE (class in lightning_ir.loss.loss) ConstantSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) ConstantSchedulerWithQuadraticWarmup (class in lightning_ir.schedulers.schedulers) copy() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) count() (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple method) create_qrels_from_dicts() (in module lightning_ir.base.validation_utils) create_run_from_scores() (in module lightning_ir.base.validation_utils) CrossEncoderConfig (class in lightning_ir.cross_encoder.cross_encoder_config) CrossEncoderModel (class in lightning_ir.cross_encoder.cross_encoder_model) CrossEncoderModule (class in lightning_ir.cross_encoder.cross_encoder_module) CrossEncoderOutput (class in lightning_ir.cross_encoder.cross_encoder_model) CrossEncoderTokenizer (class in lightning_ir.cross_encoder.cross_encoder_tokenizer)

D

DASHED_DATASET_MAP (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) dataset (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) dataset_id (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) device (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding property) doc_embeddings (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput attribute) doc_ids (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple attribute)

DOC_TOKEN (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer attribute) (lightning_ir.models.col.ColTokenizer attribute) doc_token_id (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer property) (lightning_ir.models.col.ColTokenizer property) DocDataset (class in lightning_ir.data.dataset) docs (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) docs_dataset_id (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) DocSample (class in lightning_ir.data.data) download_artifact() (lightning_ir.main.LightningIRWandbLogger static method) DprConfig (class in lightning_ir.models.dpr) DprModel (class in lightning_ir.models.dpr)

E

EmbeddingLossFunction (class in lightning_ir.loss.loss) embeddings (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding attribute) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput attribute) encode() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.splade.SpladeModel method) encode_doc() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.splade.SpladeModel method)

encode_query() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.splade.SpladeModel method) encoding (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding attribute) evaluate_run() (in module lightning_ir.base.validation_utils) ExactSearchConfig (class in lightning_ir.retrieve.base.searcher) ExactSearcher (class in lightning_ir.retrieve.base.searcher) experiment (lightning_ir.main.LightningIRWandbLogger property) extra_repr() (lightning_ir.models.t5_cross_encoder.ScaleLinear method)

F

FaissFlatIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissFlatIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIVFIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIVFIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIVFPQIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIVFPQIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissPQIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissPQIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissSearchConfig (class in lightning_ir.retrieve.faiss.faiss_searcher) FaissSearcher (class in lightning_ir.retrieve.faiss.faiss_searcher) finalize() (lightning_ir.main.LightningIRWandbLogger method) FLOPSRegularization (class in lightning_ir.loss.loss) forward() (lightning_ir.base.model.LightningIRModel method) (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderModel method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.set_encoder.SetEncoderModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.t5_cross_encoder.ScaleLinear method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderModel method) from_backbone_class() (lightning_ir.base.class_factory.LightningIRClassFactory method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory method) (lightning_ir.base.class_factory.LightningIRModelClassFactory method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory method) from_backbone_classes() (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory method) from_ir_dataset_sample() (lightning_ir.data.data.DocSample class method) (lightning_ir.data.data.QuerySample class method)

from_pretrained() (lightning_ir.base.class_factory.LightningIRClassFactory method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory method) (lightning_ir.base.class_factory.LightningIRModelClassFactory method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory method) (lightning_ir.base.config.LightningIRConfig class method) (lightning_ir.base.model.LightningIRModel class method) (lightning_ir.base.tokenizer.LightningIRTokenizer class method) (lightning_ir.bi_encoder.bi_encoder_config.BiEncoderConfig class method) (lightning_ir.bi_encoder.bi_encoder_config.MultiVectorBiEncoderConfig class method) (lightning_ir.bi_encoder.bi_encoder_config.SingleVectorBiEncoderConfig class method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel class method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel class method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel class method) (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer class method) (lightning_ir.cross_encoder.cross_encoder_config.CrossEncoderConfig class method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderModel class method) (lightning_ir.cross_encoder.cross_encoder_tokenizer.CrossEncoderTokenizer class method) (lightning_ir.models.col.ColConfig class method) (lightning_ir.models.col.ColModel class method) (lightning_ir.models.col.ColTokenizer class method) (lightning_ir.models.dpr.DprConfig class method) (lightning_ir.models.dpr.DprModel class method) (lightning_ir.models.set_encoder.SetEncoderConfig class method) (lightning_ir.models.set_encoder.SetEncoderModel class method) (lightning_ir.models.set_encoder.SetEncoderTokenizer class method) (lightning_ir.models.splade.SpladeConfig class method) (lightning_ir.models.splade.SpladeModel class method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderConfig class method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderModel class method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderTokenizer class method) fromkeys() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method)

G

GenericConstantSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) GenericConstantSchedulerWithQuadraticWarmup (class in lightning_ir.schedulers.schedulers) GenericLinearSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) GenericScheduler (class in lightning_ir.schedulers.schedulers) get() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) get_backbone_config() (lightning_ir.base.class_factory.LightningIRClassFactory static method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory static method) (lightning_ir.base.class_factory.LightningIRModelClassFactory static method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory static method) get_backbone_model_type() (lightning_ir.base.class_factory.LightningIRClassFactory static method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory static method) (lightning_ir.base.class_factory.LightningIRModelClassFactory static method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory static method) get_dataset() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) get_dataset_id() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) get_extended_attention_mask() (lightning_ir.models.set_encoder.SetEncoderModel method) get_last_lr() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method)

get_lightning_ir_config() (lightning_ir.base.class_factory.LightningIRClassFactory static method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory static method) (lightning_ir.base.class_factory.LightningIRModelClassFactory static method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory static method) get_lightning_ir_model_type() (lightning_ir.base.class_factory.LightningIRClassFactory static method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory static method) (lightning_ir.base.class_factory.LightningIRModelClassFactory static method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory static method) get_lr() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) get_output_embeddings() (lightning_ir.models.splade.SpladeModel method) get_tokenizer_kwargs() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.bi_encoder_config.BiEncoderConfig method) (lightning_ir.bi_encoder.bi_encoder_config.MultiVectorBiEncoderConfig method) (lightning_ir.bi_encoder.bi_encoder_config.SingleVectorBiEncoderConfig method) (lightning_ir.cross_encoder.cross_encoder_config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.dpr.DprConfig method) (lightning_ir.models.set_encoder.SetEncoderConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderConfig method) group_separator (lightning_ir.main.LightningIRWandbLogger property)

I

ids (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding attribute) InBatchCrossEntropy (class in lightning_ir.loss.loss) InBatchLossFunction (class in lightning_ir.loss.loss) index() (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple method) (lightning_ir.main.LightningIRTrainer method) IndexBatch (class in lightning_ir.data.data) IndexCallback (class in lightning_ir.callbacks.callbacks) IndexConfig (class in lightning_ir.retrieve.base.indexer) Indexer (class in lightning_ir.retrieve.base.indexer) indexer_class (lightning_ir.retrieve.faiss.faiss_indexer.FaissFlatIndexConfig attribute) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexConfig attribute) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexConfig attribute) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFPQIndexConfig attribute) (lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexConfig attribute) (lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexConfig attribute) (lightning_ir.retrieve.pytorch.dense_indexer.TorchDenseIndexConfig attribute) (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexConfig attribute) (lightning_ir.retrieve.seismic.seismic_indexer.SeismicIndexConfig attribute)

inference_dataloader() (lightning_ir.data.datamodule.LightningIRDataModule method) InfoNCE (class in lightning_ir.loss.loss) ir_dataset (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) IRDataset (class in lightning_ir.data.dataset) items() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method)

K

keys() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method)

KLDivergence (class in lightning_ir.loss.loss)

L

L1Regularization (class in lightning_ir.loss.loss) L2Regularization (class in lightning_ir.loss.loss) LambdaWarmupScheduler (class in lightning_ir.schedulers.schedulers) lightning_ir module lightning_ir.base module lightning_ir.base.class_factory module lightning_ir.base.config module lightning_ir.base.external_model_hub module lightning_ir.base.model module lightning_ir.base.module module lightning_ir.base.tokenizer module lightning_ir.base.validation_utils module lightning_ir.bi_encoder module lightning_ir.bi_encoder.bi_encoder_config module lightning_ir.bi_encoder.bi_encoder_model module lightning_ir.bi_encoder.bi_encoder_module module lightning_ir.bi_encoder.bi_encoder_tokenizer module lightning_ir.callbacks module lightning_ir.callbacks.callbacks module lightning_ir.cross_encoder module lightning_ir.cross_encoder.cross_encoder_config module lightning_ir.cross_encoder.cross_encoder_model module lightning_ir.cross_encoder.cross_encoder_module module lightning_ir.cross_encoder.cross_encoder_tokenizer module lightning_ir.data module lightning_ir.data.data module lightning_ir.data.datamodule module lightning_ir.data.dataset module lightning_ir.data.external_datasets module lightning_ir.data.external_datasets.colbert module lightning_ir.data.external_datasets.cross_architecture_knowledge_distillation module lightning_ir.data.external_datasets.ir_datasets_utils module lightning_ir.data.external_datasets.rank_distillm module lightning_ir.loss module lightning_ir.loss.loss module lightning_ir.main module lightning_ir.modeling_utils module lightning_ir.modeling_utils.batching module lightning_ir.modeling_utils.mlm_head module lightning_ir.models module lightning_ir.models.col module lightning_ir.models.dpr module lightning_ir.models.set_encoder module lightning_ir.models.splade module

lightning_ir.models.t5_cross_encoder module lightning_ir.register_external_models module lightning_ir.register_internal_models module lightning_ir.retrieve module lightning_ir.retrieve.base module lightning_ir.retrieve.base.indexer module lightning_ir.retrieve.base.packed_tensor module lightning_ir.retrieve.base.searcher module lightning_ir.retrieve.faiss module lightning_ir.retrieve.faiss.faiss_indexer module lightning_ir.retrieve.faiss.faiss_searcher module lightning_ir.retrieve.plaid module lightning_ir.retrieve.plaid.plaid_indexer module lightning_ir.retrieve.plaid.plaid_searcher module lightning_ir.retrieve.plaid.residual_codec module lightning_ir.retrieve.pytorch module lightning_ir.retrieve.pytorch.dense_indexer module lightning_ir.retrieve.pytorch.dense_searcher module lightning_ir.retrieve.pytorch.sparse_indexer module lightning_ir.retrieve.pytorch.sparse_searcher module lightning_ir.retrieve.seismic module lightning_ir.retrieve.seismic.seismic_format module lightning_ir.retrieve.seismic.seismic_indexer module lightning_ir.retrieve.seismic.seismic_searcher module lightning_ir.schedulers module lightning_ir.schedulers.lr_schedulers module lightning_ir.schedulers.schedulers module LightningIRClassFactory (class in lightning_ir.base.class_factory) LightningIRCLI (class in lightning_ir.main) LightningIRConfig (class in lightning_ir.base.config) LightningIRConfigClassFactory (class in lightning_ir.base.class_factory) LightningIRDataModule (class in lightning_ir.data.datamodule) LightningIRModel (class in lightning_ir.base.model) LightningIRModelClassFactory (class in lightning_ir.base.class_factory) LightningIRModule (class in lightning_ir.base.module) LightningIROutput (class in lightning_ir.base.model) LightningIRSaveConfigCallback (class in lightning_ir.main) LightningIRTokenizer (class in lightning_ir.base.tokenizer) LightningIRTokenizerClassFactory (class in lightning_ir.base.class_factory) LightningIRTrainer (class in lightning_ir.main) LightningIRWandbLogger (class in lightning_ir.main) LinearLRSchedulerWithLinearWarmup (class in lightning_ir.schedulers.lr_schedulers) LinearSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) ListwiseLossFunction (class in lightning_ir.loss.loss) load_state_dict() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) log_audio() (lightning_ir.main.LightningIRWandbLogger method) log_dir (lightning_ir.main.LightningIRWandbLogger property) log_graph() (lightning_ir.main.LightningIRWandbLogger method) log_hyperparams() (lightning_ir.main.LightningIRWandbLogger method) log_image() (lightning_ir.main.LightningIRWandbLogger method) log_metrics() (lightning_ir.main.LightningIRWandbLogger method) log_random() (lightning_ir.data.dataset.Sampler static method) log_table() (lightning_ir.main.LightningIRWandbLogger method) log_text() (lightning_ir.main.LightningIRWandbLogger method) log_video() (lightning_ir.main.LightningIRWandbLogger method) LossFunction (class in lightning_ir.loss.loss)

M

main() (in module lightning_ir.main) MarginMSE (class in lightning_ir.loss.loss) model_type (lightning_ir.base.config.LightningIRConfig attribute) (lightning_ir.bi_encoder.bi_encoder_config.BiEncoderConfig attribute) (lightning_ir.bi_encoder.bi_encoder_config.MultiVectorBiEncoderConfig attribute) (lightning_ir.bi_encoder.bi_encoder_config.SingleVectorBiEncoderConfig attribute) (lightning_ir.cross_encoder.cross_encoder_config.CrossEncoderConfig attribute) (lightning_ir.models.col.ColConfig attribute) (lightning_ir.models.dpr.DprConfig attribute) (lightning_ir.models.set_encoder.SetEncoderConfig attribute) (lightning_ir.models.splade.SpladeConfig attribute) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderConfig attribute) module lightning_ir lightning_ir.base lightning_ir.base.class_factory lightning_ir.base.config lightning_ir.base.external_model_hub lightning_ir.base.model lightning_ir.base.module lightning_ir.base.tokenizer lightning_ir.base.validation_utils lightning_ir.bi_encoder lightning_ir.bi_encoder.bi_encoder_config lightning_ir.bi_encoder.bi_encoder_model lightning_ir.bi_encoder.bi_encoder_module lightning_ir.bi_encoder.bi_encoder_tokenizer lightning_ir.callbacks lightning_ir.callbacks.callbacks lightning_ir.cross_encoder lightning_ir.cross_encoder.cross_encoder_config lightning_ir.cross_encoder.cross_encoder_model lightning_ir.cross_encoder.cross_encoder_module lightning_ir.cross_encoder.cross_encoder_tokenizer lightning_ir.data lightning_ir.data.data lightning_ir.data.datamodule lightning_ir.data.dataset lightning_ir.data.external_datasets lightning_ir.data.external_datasets.colbert lightning_ir.data.external_datasets.cross_architecture_knowledge_distillation lightning_ir.data.external_datasets.ir_datasets_utils lightning_ir.data.external_datasets.rank_distillm lightning_ir.loss lightning_ir.loss.loss lightning_ir.main lightning_ir.modeling_utils lightning_ir.modeling_utils.batching lightning_ir.modeling_utils.mlm_head lightning_ir.models lightning_ir.models.col lightning_ir.models.dpr lightning_ir.models.set_encoder lightning_ir.models.splade lightning_ir.models.t5_cross_encoder lightning_ir.register_external_models lightning_ir.register_internal_models lightning_ir.retrieve lightning_ir.retrieve.base lightning_ir.retrieve.base.indexer lightning_ir.retrieve.base.packed_tensor lightning_ir.retrieve.base.searcher lightning_ir.retrieve.faiss lightning_ir.retrieve.faiss.faiss_indexer lightning_ir.retrieve.faiss.faiss_searcher lightning_ir.retrieve.plaid lightning_ir.retrieve.plaid.plaid_indexer lightning_ir.retrieve.plaid.plaid_searcher lightning_ir.retrieve.plaid.residual_codec lightning_ir.retrieve.pytorch lightning_ir.retrieve.pytorch.dense_indexer lightning_ir.retrieve.pytorch.dense_searcher lightning_ir.retrieve.pytorch.sparse_indexer lightning_ir.retrieve.pytorch.sparse_searcher lightning_ir.retrieve.seismic lightning_ir.retrieve.seismic.seismic_format lightning_ir.retrieve.seismic.seismic_indexer lightning_ir.retrieve.seismic.seismic_searcher lightning_ir.schedulers lightning_ir.schedulers.lr_schedulers lightning_ir.schedulers.schedulers

move_to_end() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) MultiVectorBiEncoderConfig (class in lightning_ir.bi_encoder.bi_encoder_config) MultiVectorBiEncoderModel (class in lightning_ir.bi_encoder.bi_encoder_model)

N

name (lightning_ir.main.LightningIRWandbLogger property)

NeuralLossFunction (class in lightning_ir.loss.loss) num_docs (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple attribute)

O

on_save_checkpoint() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) on_test_batch_end() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) on_test_batch_start() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) on_test_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) on_test_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) on_train_batch_start() (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method)

on_train_end() (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_train_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_validation_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) on_validation_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method)

P

PackedTensor (class in lightning_ir.retrieve.base.packed_tensor) PairwiseLossFunction (class in lightning_ir.loss.loss) ParquetScoredDocs (class in lightning_ir.data.external_datasets.ir_datasets_utils) PearsonCorrelation (class in lightning_ir.loss.loss) PlaidIndexConfig (class in lightning_ir.retrieve.plaid.plaid_indexer) PlaidIndexer (class in lightning_ir.retrieve.plaid.plaid_indexer) PlaidSearchConfig (class in lightning_ir.retrieve.plaid.plaid_searcher) PlaidSearcher (class in lightning_ir.retrieve.plaid.plaid_searcher) pooling() (lightning_ir.base.model.LightningIRModel method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.set_encoder.SetEncoderModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderModel method) pop() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method)

popitem() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) POST_LOAD_CALLBACKS (in module lightning_ir.base.external_model_hub) predict_dataloader() (lightning_ir.data.datamodule.LightningIRDataModule method) prepare_constituent() (lightning_ir.data.dataset.DocDataset method) (lightning_ir.data.dataset.IRDataset method) (lightning_ir.data.dataset.QueryDataset method) (lightning_ir.data.dataset.RunDataset method) (lightning_ir.data.dataset.TupleDataset method) prepare_data() (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.data.dataset.DocDataset method) (lightning_ir.data.dataset.QueryDataset method) (lightning_ir.data.dataset.RunDataset method) (lightning_ir.data.dataset.TupleDataset method) prepare_input() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method)

Q

qrels (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) queries (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property)

query_embeddings (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput attribute) query_id (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple attribute) QUERY_TOKEN (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer attribute) (lightning_ir.models.col.ColTokenizer attribute) query_token_id (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer property) (lightning_ir.models.col.ColTokenizer property) QueryDataset (class in lightning_ir.data.dataset) QuerySample (class in lightning_ir.data.data)

R

random() (lightning_ir.data.dataset.Sampler static method) RankBatch (class in lightning_ir.data.data) RankCallback (class in lightning_ir.callbacks.callbacks) RankNet (class in lightning_ir.loss.loss) RankSample (class in lightning_ir.data.data) re_rank() (lightning_ir.main.LightningIRTrainer method) register_colbert_docpairs() (in module lightning_ir.data.external_datasets.colbert) register_kd_docpairs() (in module lightning_ir.data.external_datasets.cross_architecture_knowledge_distillation)

register_new_dataset() (in module lightning_ir.data.external_datasets.ir_datasets_utils) register_rank_distillm() (in module lightning_ir.data.external_datasets.rank_distillm) RegisterLocalDatasetCallback (class in lightning_ir.callbacks.callbacks) RegularizationLossFunction (class in lightning_ir.loss.loss) ReRankCallback (class in lightning_ir.callbacks.callbacks) ResidualCodec (class in lightning_ir.retrieve.plaid.residual_codec) root_dir (lightning_ir.main.LightningIRWandbLogger property) RunDataset (class in lightning_ir.data.dataset)

S

sample() (lightning_ir.data.dataset.Sampler static method) Sampler (class in lightning_ir.data.dataset) save() (lightning_ir.main.LightningIRWandbLogger method) save_config() (lightning_ir.main.LightningIRSaveConfigCallback method) save_dir (lightning_ir.main.LightningIRWandbLogger property) save_pretrained() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) ScaleLinear (class in lightning_ir.models.t5_cross_encoder) score() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.splade.SpladeModel method) ScoreBasedInBatchCrossEntropy (class in lightning_ir.loss.loss) ScoreBasedInBatchLossFunction (class in lightning_ir.loss.loss) ScoredDocTuple (class in lightning_ir.data.external_datasets.ir_datasets_utils) ScoredDocTuples (class in lightning_ir.data.external_datasets.ir_datasets_utils) scores (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple attribute) scoring_mask (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding attribute) scoring_mask() (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.models.col.ColModel method) ScoringLossFunction (class in lightning_ir.loss.loss) search() (lightning_ir.main.LightningIRTrainer method) search_class (lightning_ir.retrieve.base.searcher.ApproximateSearchConfig attribute) (lightning_ir.retrieve.base.searcher.ExactSearchConfig attribute) (lightning_ir.retrieve.faiss.faiss_searcher.FaissSearchConfig attribute) (lightning_ir.retrieve.plaid.plaid_searcher.PlaidSearchConfig attribute) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseSearchConfig attribute) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseSearchConfig attribute) (lightning_ir.retrieve.seismic.seismic_searcher.SeismicSearchConfig attribute) SearchBatch (class in lightning_ir.data.data) SearchCallback (class in lightning_ir.callbacks.callbacks) SearchConfig (class in lightning_ir.retrieve.base.searcher) Searcher (class in lightning_ir.retrieve.base.searcher) searcher (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule property) SeismicFormatConverter (class in lightning_ir.retrieve.seismic.seismic_format) SeismicIndexConfig (class in lightning_ir.retrieve.seismic.seismic_indexer)

SeismicIndexer (class in lightning_ir.retrieve.seismic.seismic_indexer) SeismicSearchConfig (class in lightning_ir.retrieve.seismic.seismic_searcher) SeismicSearcher (class in lightning_ir.retrieve.seismic.seismic_searcher) set_optimizer() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) setdefault() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) SetEncoderConfig (class in lightning_ir.models.set_encoder) SetEncoderModel (class in lightning_ir.models.set_encoder) SetEncoderTokenizer (class in lightning_ir.models.set_encoder) setup() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) single_relevant() (lightning_ir.data.dataset.Sampler static method) SingleVectorBiEncoderConfig (class in lightning_ir.bi_encoder.bi_encoder_config) SingleVectorBiEncoderModel (class in lightning_ir.bi_encoder.bi_encoder_model) sparsification() (lightning_ir.base.model.LightningIRModel method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method) (lightning_ir.bi_encoder.bi_encoder_model.SingleVectorBiEncoderModel method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.dpr.DprModel method) (lightning_ir.models.set_encoder.SetEncoderModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderModel method) SpladeConfig (class in lightning_ir.models.splade) SpladeModel (class in lightning_ir.models.splade) state_dict() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) STATE_DICT_KEY_MAPPING (in module lightning_ir.base.external_model_hub) step() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) subcommands() (lightning_ir.main.LightningIRCLI static method) SupervisedMarginMSE (class in lightning_ir.loss.loss)

T

T5CrossEncoderConfig (class in lightning_ir.models.t5_cross_encoder) T5CrossEncoderModel (class in lightning_ir.models.t5_cross_encoder) T5CrossEncoderTokenizer (class in lightning_ir.models.t5_cross_encoder) teardown() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) test_dataloader() (lightning_ir.data.datamodule.LightningIRDataModule method) test_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) to() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding method) to_dict() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.bi_encoder_config.BiEncoderConfig method) (lightning_ir.bi_encoder.bi_encoder_config.MultiVectorBiEncoderConfig method) (lightning_ir.bi_encoder.bi_encoder_config.SingleVectorBiEncoderConfig method) (lightning_ir.cross_encoder.cross_encoder_config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.dpr.DprConfig method) (lightning_ir.models.set_encoder.SetEncoderConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderConfig method) to_padded_tensor() (lightning_ir.retrieve.base.packed_tensor.PackedTensor method) to_tuple() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) tokenize() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.cross_encoder_tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.col.ColTokenizer method) (lightning_ir.models.set_encoder.SetEncoderTokenizer method) (lightning_ir.models.t5_cross_encoder.T5CrossEncoderTokenizer method)

tokenize_doc() (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer method) (lightning_ir.models.col.ColTokenizer method) tokenize_input_sequence() (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer method) (lightning_ir.models.col.ColTokenizer method) tokenize_query() (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer method) (lightning_ir.models.col.ColTokenizer method) top() (lightning_ir.data.dataset.Sampler static method) top_and_random() (lightning_ir.data.dataset.Sampler static method) TorchDenseIndex (class in lightning_ir.retrieve.pytorch.dense_searcher) TorchDenseIndexConfig (class in lightning_ir.retrieve.pytorch.dense_indexer) TorchDenseIndexer (class in lightning_ir.retrieve.pytorch.dense_indexer) TorchDenseSearchConfig (class in lightning_ir.retrieve.pytorch.dense_searcher) TorchDenseSearcher (class in lightning_ir.retrieve.pytorch.dense_searcher) TorchSparseIndex (class in lightning_ir.retrieve.pytorch.sparse_searcher) TorchSparseIndexConfig (class in lightning_ir.retrieve.pytorch.sparse_indexer) TorchSparseIndexer (class in lightning_ir.retrieve.pytorch.sparse_indexer) TorchSparseSearchConfig (class in lightning_ir.retrieve.pytorch.sparse_searcher) TorchSparseSearcher (class in lightning_ir.retrieve.pytorch.sparse_searcher) train_dataloader() (lightning_ir.data.datamodule.LightningIRDataModule method) TrainBatch (class in lightning_ir.data.data) training_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) TupleDataset (class in lightning_ir.data.dataset)

U

update() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method)

use_artifact() (lightning_ir.main.LightningIRWandbLogger method)

V

val_dataloader() (lightning_ir.data.datamodule.LightningIRDataModule method) validate() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) validate_loss() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) validate_metrics() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) validation_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.bi_encoder_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method)

value_lambda() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) (lightning_ir.schedulers.schedulers.ConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.ConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.LambdaWarmupScheduler method) (lightning_ir.schedulers.schedulers.LinearSchedulerWithLinearWarmup method) values() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput method) version (lightning_ir.main.LightningIRWandbLogger property)

W

WarmupLRScheduler (class in lightning_ir.schedulers.lr_schedulers)