Index

_

__init__() (lightning_ir.base.adapter.LightningIRAdapterMixin method) (lightning_ir.base.class_factory.LightningIRClassFactory 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.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.TupleDataset method) (lightning_ir.data.external_datasets.ir_datasets_utils.ParquetScoredDocs method) (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuples method) (lightning_ir.loss.approximate.ApproxLossFunction method) (lightning_ir.loss.approximate.ApproxMRR method) (lightning_ir.loss.approximate.ApproxNDCG method) (lightning_ir.loss.approximate.ApproxRankMSE method) (lightning_ir.loss.base.InBatchLossFunction method) (lightning_ir.loss.base.RegularizationLossFunction method) (lightning_ir.loss.embedding.ContrastiveLocalLoss method) (lightning_ir.loss.in_batch.ScoreBasedInBatchLossFunction method) (lightning_ir.loss.neural.NeuralLossFunction method) (lightning_ir.loss.pairwise.ConstantMarginMSE method) (lightning_ir.loss.pairwise.MarginMSE method) (lightning_ir.loss.pairwise.RankNet method) (lightning_ir.loss.pairwise.SupervisedMarginMSE method) (lightning_ir.modeling_utils.mlm_head.DistilBertOnlyMLMHead method) (lightning_ir.models.bi_encoders.coil.CoilConfig method) (lightning_ir.models.bi_encoders.coil.CoilEmbedding method) (lightning_ir.models.bi_encoders.coil.CoilModel method) (lightning_ir.models.bi_encoders.coil.CoilOutput method) (lightning_ir.models.bi_encoders.coil.UniCoilConfig method) (lightning_ir.models.bi_encoders.coil.UniCoilModel method) (lightning_ir.models.bi_encoders.col.ColConfig method) (lightning_ir.models.bi_encoders.col.ColModel method) (lightning_ir.models.bi_encoders.col.ColTokenizer method) (lightning_ir.models.bi_encoders.dpr.DprConfig method) (lightning_ir.models.bi_encoders.dpr.DprModel method) (lightning_ir.models.bi_encoders.mvr.MvrConfig method) (lightning_ir.models.bi_encoders.mvr.MvrModel method) (lightning_ir.models.bi_encoders.mvr.MvrTokenizer method) (lightning_ir.models.bi_encoders.splade.SpladeConfig method) (lightning_ir.models.bi_encoders.splade.SpladeModel method) (lightning_ir.models.bi_encoders.splade.SpladeTokenizer method) (lightning_ir.models.cross_encoders.mono.MonoConfig method) (lightning_ir.models.cross_encoders.mono.MonoModel method) (lightning_ir.models.cross_encoders.set_encoder.SetEncoderConfig method) (lightning_ir.models.cross_encoders.set_encoder.SetEncoderModel method) (lightning_ir.models.cross_encoders.set_encoder.SetEncoderTokenizer 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.SearchConfig method) (lightning_ir.retrieve.base.searcher.Searcher method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissFlatIndexer 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.pytorch.dense_indexer.TorchDenseIndexer method) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseIndex method) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseSearcher method) (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexer method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseIndex method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseSearcher 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.WarmupLRScheduler method) (lightning_ir.schedulers.schedulers.GenericScheduler method) (lightning_ir.schedulers.schedulers.LambdaWarmupScheduler method) (lightning_ir.schedulers.schedulers.LinearSchedulerWithLinearWarmup method)

A

add() (lightning_ir.retrieve.base.indexer.Indexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexer method) (lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexer method) (lightning_ir.retrieve.pytorch.dense_indexer.TorchDenseIndexer method) (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexer method) (lightning_ir.retrieve.seismic.seismic_indexer.SeismicIndexer method) add_arguments_to_parser() (lightning_ir.main.LightningIRCLI method) aggregate_similarity() (lightning_ir.bi_encoder.bi_encoder_model.MultiVectorBiEncoderModel method)

ALLOW_SUB_BATCHING (lightning_ir.base.model.LightningIRModel attribute), [1] (lightning_ir.models.cross_encoders.set_encoder.SetEncoderModel attribute) ApproximateSearchConfig (class in lightning_ir.retrieve.base.searcher) ApproximateSearcher (class in lightning_ir.retrieve.base.searcher) ApproxLossFunction (class in lightning_ir.loss.approximate) ApproxMRR (class in lightning_ir.loss.approximate) ApproxNDCG (class in lightning_ir.loss.approximate) ApproxRankMSE (class in lightning_ir.loss.approximate) attention_forward() (lightning_ir.models.cross_encoders.set_encoder.SetEncoderModel static method)

B

BACKBONE_MAPPING (in module lightning_ir.base.external_model_hub) backbone_model_type (lightning_ir.base.config.LightningIRConfig 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) cat_other_doc_hidden_states() (lightning_ir.models.cross_encoders.set_encoder.SetEncoderModel method) cc_lir_model_type() (lightning_ir.base.class_factory.LightningIRClassFactory method) CHECKPOINT_MAPPING (in module lightning_ir.base.external_model_hub) cls_embeddings (lightning_ir.models.bi_encoders.coil.CoilEmbedding attribute) CoilConfig (class in lightning_ir.models.bi_encoders.coil) CoilEmbedding (class in lightning_ir.models.bi_encoders.coil) CoilModel (class in lightning_ir.models.bi_encoders.coil) CoilOutput (class in lightning_ir.models.bi_encoders.coil) ColConfig (class in lightning_ir.models.bi_encoders.col) ColModel (class in lightning_ir.models.bi_encoders.col) ColTokenizer (class in lightning_ir.models.bi_encoders.col) compute_loss() (lightning_ir.loss.approximate.ApproxMRR method) (lightning_ir.loss.approximate.ApproxNDCG method) (lightning_ir.loss.approximate.ApproxRankMSE method) (lightning_ir.loss.base.EmbeddingLossFunction method) (lightning_ir.loss.base.LossFunction method) (lightning_ir.loss.base.ScoringLossFunction method) (lightning_ir.loss.embedding.ContrastiveLocalLoss method) (lightning_ir.loss.in_batch.InBatchCrossEntropy method) (lightning_ir.loss.in_batch.ScoreBasedInBatchCrossEntropy method) (lightning_ir.loss.listwise.InfoNCE method) (lightning_ir.loss.listwise.KLDivergence method) (lightning_ir.loss.listwise.PearsonCorrelation method) (lightning_ir.loss.pairwise.MarginMSE method) (lightning_ir.loss.pairwise.RankNet method) (lightning_ir.loss.regularization.FLOPSRegularization method) (lightning_ir.loss.regularization.L1Regularization method) (lightning_ir.loss.regularization.L2Regularization method) compute_similarity() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) config_class (lightning_ir.base.model.LightningIRModel attribute), [1] (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.bi_encoders.coil.CoilModel attribute) (lightning_ir.models.bi_encoders.coil.UniCoilModel attribute) (lightning_ir.models.bi_encoders.col.ColModel attribute) (lightning_ir.models.bi_encoders.col.ColTokenizer attribute) (lightning_ir.models.bi_encoders.dpr.DprModel attribute) (lightning_ir.models.bi_encoders.mvr.MvrModel attribute) (lightning_ir.models.bi_encoders.mvr.MvrTokenizer attribute) (lightning_ir.models.bi_encoders.splade.SpladeModel attribute) (lightning_ir.models.bi_encoders.splade.SpladeTokenizer attribute) (lightning_ir.models.cross_encoders.mono.MonoModel attribute) (lightning_ir.models.cross_encoders.set_encoder.SetEncoderModel attribute) (lightning_ir.models.cross_encoders.set_encoder.SetEncoderTokenizer attribute)

configure_optimizers() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.main.LightningIRCLI static method) ConstantLRSchedulerWithLinearWarmup (class in lightning_ir.schedulers.lr_schedulers) ConstantMarginMSE (class in lightning_ir.loss.pairwise) ConstantSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) ConstantSchedulerWithQuadraticWarmup (class in lightning_ir.schedulers.schedulers) ContrastiveLocalLoss (class in lightning_ir.loss.embedding) convert_to_seismic_format() (lightning_ir.retrieve.seismic.seismic_format.SeismicFormatConverter static method) cosine_similarity() (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseIndex static method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseIndex static 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.IRDataset property) dashed_docs_dataset_id (lightning_ir.data.dataset.IRDataset property) dataset (lightning_ir.data.dataset.IRDataset property) dataset_id (lightning_ir.data.dataset.IRDataset property) device (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding property) disable_adapters() (lightning_ir.base.adapter.LightningIRAdapterMixin method) DistilBertOnlyMLMHead (class in lightning_ir.modeling_utils.mlm_head) doc (lightning_ir.data.data.DocSample attribute) doc_embeddings (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput attribute) (lightning_ir.models.bi_encoders.coil.CoilOutput attribute) doc_id (lightning_ir.data.data.DocSample attribute) doc_ids (lightning_ir.data.data.IndexBatch attribute) (lightning_ir.data.data.RankBatch attribute) (lightning_ir.data.data.RankSample attribute) (lightning_ir.data.data.SearchBatch attribute) (lightning_ir.data.data.TrainBatch attribute) (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple attribute)

DOC_TOKEN (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer attribute) doc_token_id (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer property) doc_token_idcs (lightning_ir.retrieve.base.searcher.ExactSearcher property) DocDataset (class in lightning_ir.data.dataset) docs (lightning_ir.data.data.IndexBatch attribute) (lightning_ir.data.data.RankBatch attribute) (lightning_ir.data.data.RankSample attribute) (lightning_ir.data.data.TrainBatch attribute) (lightning_ir.data.dataset.IRDataset property) docs_dataset_id (lightning_ir.data.dataset.IRDataset property) DocSample (class in lightning_ir.data.data) dot_similarity() (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseIndex static method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseIndex static method) DprConfig (class in lightning_ir.models.bi_encoders.dpr) DprModel (class in lightning_ir.models.bi_encoders.dpr)

E

EmbeddingLossFunction (class in lightning_ir.loss.base) embeddings (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding attribute) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput attribute) (lightning_ir.models.bi_encoders.coil.CoilEmbedding attribute) enable_adapters() (lightning_ir.base.adapter.LightningIRAdapterMixin method) encode() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) (lightning_ir.models.bi_encoders.coil.CoilModel method) (lightning_ir.models.bi_encoders.coil.UniCoilModel method) (lightning_ir.models.bi_encoders.col.ColModel method) (lightning_ir.models.bi_encoders.dpr.DprModel method) (lightning_ir.models.bi_encoders.mvr.MvrModel method) (lightning_ir.models.bi_encoders.splade.SpladeModel method)

encode_doc() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel method) encode_query() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderModel 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)

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) FLOPSRegularization (class in lightning_ir.loss.regularization) 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_module.BiEncoderModule method) (lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderModel method) (lightning_ir.cross_encoder.cross_encoder_module.CrossEncoderModule method) (lightning_ir.modeling_utils.mlm_head.DistilBertOnlyMLMHead method) (lightning_ir.models.cross_encoders.mono.MonoModel method) (lightning_ir.models.cross_encoders.mono.ScaleLinear method) (lightning_ir.models.cross_encoders.set_encoder.SetEncoderModel 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.models.bi_encoders.splade.SpladeModel class method) (lightning_ir.retrieve.base.indexer.IndexConfig class 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_approx_ranks() (lightning_ir.loss.approximate.ApproxLossFunction static method) get_backbone_config() (lightning_ir.base.class_factory.LightningIRClassFactory 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.LightningIRTokenizerClassFactory static method) get_dataset() (lightning_ir.base.module.LightningIRModule method) get_dataset_id() (lightning_ir.base.module.LightningIRModule method)

get_dcg() (lightning_ir.loss.approximate.ApproxNDCG static method) get_extended_attention_mask() (lightning_ir.models.cross_encoders.set_encoder.SetEncoderModel method) get_ib_idcs() (lightning_ir.loss.base.InBatchLossFunction method) get_lightning_ir_config() (lightning_ir.base.class_factory.LightningIRClassFactory static method) get_lightning_ir_model_type() (lightning_ir.base.class_factory.LightningIRClassFactory static method) get_mrr() (lightning_ir.loss.approximate.ApproxMRR static method) get_ndcg() (lightning_ir.loss.approximate.ApproxNDCG static method) get_output_embeddings() (lightning_ir.models.bi_encoders.splade.SpladeModel method) get_pairwise_idcs() (lightning_ir.loss.base.PairwiseLossFunction method) get_sorted_targets() (lightning_ir.loss.neural.NeuralLossFunction method) get_tokenizer_kwargs() (lightning_ir.base.config.LightningIRConfig method)

I

ids (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding attribute) InBatchCrossEntropy (class in lightning_ir.loss.in_batch) InBatchLossFunction (class in lightning_ir.loss.base) index() (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.listwise) init_adapters() (lightning_ir.base.adapter.LightningIRAdapterMixin method) ir_dataset (lightning_ir.data.dataset.IRDataset property) IRDataset (class in lightning_ir.data.dataset) items() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding method)

K

KLDivergence (class in lightning_ir.loss.listwise)

L

L1Regularization (class in lightning_ir.loss.regularization) L2Regularization (class in lightning_ir.loss.regularization) LambdaWarmupScheduler (class in lightning_ir.schedulers.schedulers) lightning_ir module lightning_ir.base module lightning_ir.base.adapter 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.approximate module lightning_ir.loss.base module lightning_ir.loss.embedding module lightning_ir.loss.in_batch module lightning_ir.loss.listwise module lightning_ir.loss.neural module lightning_ir.loss.pairwise module lightning_ir.loss.regularization 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.bi_encoders module lightning_ir.models.bi_encoders.coil module lightning_ir.models.bi_encoders.col module lightning_ir.models.bi_encoders.dpr module lightning_ir.models.bi_encoders.mvr module lightning_ir.models.bi_encoders.splade module lightning_ir.models.cross_encoders module lightning_ir.models.cross_encoders.mono module lightning_ir.models.cross_encoders.set_encoder module lightning_ir.models.register_external_models module lightning_ir.models.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.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 LightningIRAdapterMixin (class in lightning_ir.base.adapter) 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.base) log_random() (lightning_ir.data.dataset.Sampler static method) lookup() (lightning_ir.retrieve.base.packed_tensor.PackedTensor method) LossFunction (class in lightning_ir.loss.base)

M

main() (in module lightning_ir.main) MarginMSE (class in lightning_ir.loss.pairwise) 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.bi_encoders.coil.CoilConfig attribute) (lightning_ir.models.bi_encoders.coil.UniCoilConfig attribute) (lightning_ir.models.bi_encoders.col.ColConfig attribute) (lightning_ir.models.bi_encoders.dpr.DprConfig attribute) (lightning_ir.models.bi_encoders.mvr.MvrConfig attribute) (lightning_ir.models.bi_encoders.splade.SpladeConfig attribute) (lightning_ir.models.cross_encoders.mono.MonoConfig attribute) (lightning_ir.models.cross_encoders.set_encoder.SetEncoderConfig attribute) module lightning_ir lightning_ir.base lightning_ir.base.adapter 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.approximate lightning_ir.loss.base lightning_ir.loss.embedding lightning_ir.loss.in_batch lightning_ir.loss.listwise lightning_ir.loss.neural lightning_ir.loss.pairwise lightning_ir.loss.regularization lightning_ir.main lightning_ir.modeling_utils lightning_ir.modeling_utils.batching lightning_ir.modeling_utils.mlm_head lightning_ir.models lightning_ir.models.bi_encoders lightning_ir.models.bi_encoders.coil lightning_ir.models.bi_encoders.col lightning_ir.models.bi_encoders.dpr lightning_ir.models.bi_encoders.mvr lightning_ir.models.bi_encoders.splade lightning_ir.models.cross_encoders lightning_ir.models.cross_encoders.mono lightning_ir.models.cross_encoders.set_encoder lightning_ir.models.register_external_models lightning_ir.models.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.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

MonoConfig (class in lightning_ir.models.cross_encoders.mono) MonoModel (class in lightning_ir.models.cross_encoders.mono) MultiVectorBiEncoderConfig (class in lightning_ir.bi_encoder.bi_encoder_config) MultiVectorBiEncoderModel (class in lightning_ir.bi_encoder.bi_encoder_model) MvrConfig (class in lightning_ir.models.bi_encoders.mvr) MvrModel (class in lightning_ir.models.bi_encoders.mvr) MvrTokenizer (class in lightning_ir.models.bi_encoders.mvr)

N

neural_sort() (lightning_ir.loss.neural.NeuralLossFunction method) NeuralLossFunction (class in lightning_ir.loss.neural)

num_docs (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple attribute) num_embeddings (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseIndex property) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseIndex property)

O

on_save_checkpoint() (lightning_ir.base.module.LightningIRModule method) on_test_batch_end() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback 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) 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)

on_train_batch_start() (lightning_ir.schedulers.schedulers.GenericScheduler method) on_train_end() (lightning_ir.schedulers.schedulers.GenericScheduler method) on_train_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_validation_end() (lightning_ir.base.module.LightningIRModule method) on_validation_start() (lightning_ir.base.module.LightningIRModule method)

P

PackedTensor (class in lightning_ir.retrieve.base.packed_tensor) PairwiseLossFunction (class in lightning_ir.loss.base) ParquetScoredDocs (class in lightning_ir.data.external_datasets.ir_datasets_utils) PearsonCorrelation (class in lightning_ir.loss.listwise) 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) 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.IRDataset 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) process_embeddings() (lightning_ir.loss.base.RegularizationLossFunction method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexer method) process_scores() (lightning_ir.loss.base.LossFunction method) process_targets() (lightning_ir.loss.base.LossFunction method)

Q

qrels (lightning_ir.data.data.QuerySample attribute) (lightning_ir.data.data.RankBatch attribute) (lightning_ir.data.data.RankSample attribute) (lightning_ir.data.data.SearchBatch attribute) (lightning_ir.data.data.TrainBatch attribute) (lightning_ir.data.dataset.IRDataset property) (lightning_ir.data.dataset.RunDataset property) queries (lightning_ir.data.data.RankBatch attribute) (lightning_ir.data.data.SearchBatch attribute) (lightning_ir.data.data.TrainBatch attribute) (lightning_ir.data.dataset.IRDataset property) query (lightning_ir.data.data.QuerySample attribute) (lightning_ir.data.data.RankSample attribute)

query_embeddings (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput attribute) (lightning_ir.models.bi_encoders.coil.CoilOutput attribute) query_id (lightning_ir.data.data.QuerySample attribute) (lightning_ir.data.data.RankSample attribute) (lightning_ir.data.external_datasets.ir_datasets_utils.ScoredDocTuple attribute) query_ids (lightning_ir.data.data.RankBatch attribute) (lightning_ir.data.data.SearchBatch attribute) (lightning_ir.data.data.TrainBatch attribute) QUERY_TOKEN (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer attribute) query_token_id (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer 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.pairwise) 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.base) ReRankCallback (class in lightning_ir.callbacks.callbacks) 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.retrieve.base.indexer.IndexConfig method) (lightning_ir.retrieve.base.indexer.Indexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexer method) (lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexer method) (lightning_ir.retrieve.pytorch.dense_indexer.TorchDenseIndexer method) (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexer method) (lightning_ir.retrieve.seismic.seismic_indexer.SeismicIndexer method) save_dir (lightning_ir.main.LightningIRWandbLogger property) save_pretrained() (lightning_ir.base.module.LightningIRModule method) ScaleLinear (class in lightning_ir.models.cross_encoders.mono) 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.models.bi_encoders.coil.CoilModel method) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseIndex method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseIndex method) ScoreBasedInBatchCrossEntropy (class in lightning_ir.loss.in_batch) ScoreBasedInBatchLossFunction (class in lightning_ir.loss.in_batch) 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.base.model.LightningIROutput attribute) (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.bi_encoders.col.ColModel method) (lightning_ir.models.bi_encoders.mvr.MvrModel method) ScoringLossFunction (class in lightning_ir.loss.base) search() (lightning_ir.main.LightningIRTrainer method) (lightning_ir.retrieve.base.searcher.ApproximateSearcher method) (lightning_ir.retrieve.base.searcher.ExactSearcher method) (lightning_ir.retrieve.base.searcher.Searcher method) (lightning_ir.retrieve.plaid.plaid_searcher.PlaidSearcher 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) segmented_tensor (lightning_ir.retrieve.base.packed_tensor.PackedTensor 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) set_output_embeddings() (lightning_ir.models.bi_encoders.splade.SpladeModel method) set_verbosity() (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFPQIndexer method) SetEncoderConfig (class in lightning_ir.models.cross_encoders.set_encoder) SetEncoderModel (class in lightning_ir.models.cross_encoders.set_encoder) SetEncoderTokenizer (class in lightning_ir.models.cross_encoders.set_encoder) setup() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) similarity (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderOutput attribute) 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) sinkhorn_scaling() (lightning_ir.loss.neural.NeuralLossFunction method) sparsification() (lightning_ir.base.model.LightningIRModel method) SpladeConfig (class in lightning_ir.models.bi_encoders.splade) SpladeModel (class in lightning_ir.models.bi_encoders.splade) SpladeTokenizer (class in lightning_ir.models.bi_encoders.splade) STATE_DICT_KEY_MAPPING (in module lightning_ir.base.external_model_hub) subcommands() (lightning_ir.main.LightningIRCLI static method) SupervisedMarginMSE (class in lightning_ir.loss.pairwise)

T

targets (lightning_ir.data.data.RankSample attribute) (lightning_ir.data.data.TrainBatch attribute) teardown() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) test_dataloader() (lightning_ir.data.datamodule.LightningIRDataModule method) test_step() (lightning_ir.base.module.LightningIRModule method) to() (lightning_ir.bi_encoder.bi_encoder_model.BiEncoderEmbedding method) to_cpu() (lightning_ir.retrieve.faiss.faiss_indexer.FaissFlatIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexer method) (lightning_ir.retrieve.pytorch.dense_indexer.TorchDenseIndexer method) (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexer method) to_dict() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.retrieve.base.indexer.IndexConfig method) to_diff_dict() (lightning_ir.bi_encoder.bi_encoder_config.BiEncoderConfig method) to_gpu() (lightning_ir.retrieve.base.searcher.Searcher method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissFlatIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexer method) (lightning_ir.retrieve.pytorch.dense_indexer.TorchDenseIndexer method) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseIndex method) (lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseSearcher method) (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexer method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseIndex method) (lightning_ir.retrieve.pytorch.sparse_searcher.TorchSparseSearcher method)

to_padded_tensor() (lightning_ir.retrieve.base.packed_tensor.PackedTensor method) to_sparse_csr() (lightning_ir.retrieve.pytorch.sparse_indexer.TorchSparseIndexer static method) token_embeddings (lightning_ir.models.bi_encoders.coil.CoilEmbedding attribute) 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.cross_encoders.set_encoder.SetEncoderTokenizer method) tokenize_doc() (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer method) tokenize_input_sequence() (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer method) (lightning_ir.models.bi_encoders.col.ColTokenizer method) (lightning_ir.models.bi_encoders.splade.SpladeTokenizer method) tokenize_query() (lightning_ir.bi_encoder.bi_encoder_tokenizer.BiEncoderTokenizer 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) TupleDataset (class in lightning_ir.data.dataset)

U

UniCoilConfig (class in lightning_ir.models.bi_encoders.coil)

UniCoilModel (class in lightning_ir.models.bi_encoders.coil)

V

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

value_lambda() (lightning_ir.schedulers.schedulers.ConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.ConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.LambdaWarmupScheduler method) (lightning_ir.schedulers.schedulers.LinearSchedulerWithLinearWarmup method) viewer_token_id() (lightning_ir.models.bi_encoders.mvr.MvrTokenizer method)

W

WarmupLRScheduler (class in lightning_ir.schedulers.lr_schedulers)