• Trainer

Containers

• MaterializedQRelConfig
• MaterializedQRel
• ResultHeapq
• FastResultHeapq

Data

• DataArguments
• BinaryDataset
• MultiLevelDataset
• EncodingDataset
• RetrievalCollator
• file_reader
• file_reader_functions
• VectorCacheMixin

Modeling

• ModelArguments
• PretrainedEncoder
• DefaultEncoder
• ContrieverEncoder
• PretrainedRetriever
• BiEncoderRetriever
• RetrievalLoss
• InfoNCELoss

Evaluation

• EvaluationArguments
• RetrievalEvaluator
• RetrievalEvaluatorUtilsMixin
• IRMetrics
• RelevanceEvaluatorPlus
• WandbCallback