Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
☆451Mar 26, 2024Updated 2 years ago
Alternatives and similar repositories for GradCache
Users that are interested in GradCache are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Train Dense Passage Retriever (DPR) with a single GPU☆137Jun 16, 2021Updated 5 years ago
- EMNLP 2021 - Pre-training architectures for dense retrieval☆256Mar 18, 2022Updated 4 years ago
- Tevatron - Unified Document Retrieval Toolkit across Scale, Language, and Modality. Demo in SIGIR 2023, SIGIR 2025.☆750Updated this week
- NAACL2021 - COIL Contextualized Lexical Retriever☆158Jul 27, 2021Updated 5 years ago
- SIGIR 2021: Efficiently Teaching an Effective Dense Retriever with Balanced Topic Aware Sampling☆60Jul 11, 2021Updated 5 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Code Roberta version of RetroMAE: Pre-Training Retrieval-oriented Language Models Via Masked Auto-Encoder☆10Mar 16, 2023Updated 3 years ago
- Scalable training for dense retrieval models.☆297Jul 2, 2026Updated 3 months ago
- [CVPR 2025 Highlight] The official CLIP training codebase of Inf-CL: "Breaking the Memory Barrier: Near Infinite Batch Size Scaling for C…☆288Jan 16, 2025Updated last year
- Build Text Rerankers with Deep Language Models☆264Feb 20, 2024Updated 2 years ago
- Generative Representational Instruction Tuning☆702Jun 25, 2025Updated last year
- A toolkit for asynchronously validating dense retriever checkpoints during training.☆28Aug 10, 2023Updated 3 years ago
- [SIGIR 2025] The official repo for "Scaling Sparse and Dense Retrieval in Decoder-Only LLMs"☆22Mar 31, 2025Updated last year
- Codebase for RetroMAE and beyond.☆275Jun 7, 2024Updated 2 years ago
- SGPT: GPT Sentence Embeddings for Semantic Search☆870Feb 17, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Code and data of the EMNLP 2022 Main Conference paper "Reduce Catastrophic Forgetting of Dense Retrieval Training with Teleportation Nega…☆18Mar 25, 2024Updated 2 years ago
- Code for COLING22 paper, DPTDR: Deep Prompt Tuning for Dense Passage Retrieval☆26Aug 7, 2023Updated 3 years ago
- A novel embedding training algorithm leveraging ANN search and achieved SOTA retrieval on Trec DL 2019 and OpenQA benchmarks☆390Jan 6, 2026Updated 9 months ago
- SIGIR'21: Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track.☆127Feb 15, 2022Updated 4 years ago
- Pyserini is a Python toolkit for reproducible information retrieval research with sparse and dense representations.☆2,175Updated this week
- Unified Learned Sparse Retrieval Framework☆66May 13, 2024Updated 2 years ago
- A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.☆2,325Oct 16, 2025Updated last year
- ☆47Mar 27, 2022Updated 4 years ago
- ☆72Jun 16, 2022Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- This is the code repo for our paper "Learning More Effective Representations for Dense Retrieval through Deliberate Thinking Before Searc…☆29Mar 2, 2025Updated last year
- A curated list of awesome papers related to pre-trained models for information retrieval (a.k.a., pretraining for IR).☆677Jan 7, 2024Updated 2 years ago
- IntructIR, a novel benchmark specifically designed to evaluate the instruction following ability in information retrieval models. Our foc…☆32Jun 13, 2024Updated 2 years ago
- Search Engines with Autoregressive Language models☆296Apr 4, 2023Updated 3 years ago
- Dense hybrid representations for text retrieval☆65Apr 3, 2023Updated 3 years ago
- Training & evaluation library for text-based neural re-ranking and dense retrieval models built with PyTorch☆267Jan 27, 2023Updated 3 years ago
- Code release for SLIP Self-supervision meets Language-Image Pre-training☆788Feb 9, 2023Updated 3 years ago
- E5-V: Universal Embeddings with Multimodal Large Language Models☆273Dec 10, 2025Updated 10 months ago
- [EMNLP 2022] This is the code repo for our EMNLP‘22 paper "COCO-DR: Combating Distribution Shifts in Zero-Shot Dense Retrieval with Contr…☆51Oct 12, 2023Updated 3 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Contriever: Unsupervised Dense Information Retrieval with Contrastive Learning☆779Apr 7, 2023Updated 3 years ago
- Train Models Contrastively in Pytorch☆805Mar 26, 2025Updated last year
- Dense Passage Retriever - is a set of tools and models for open domain Q&A task.☆1,869Apr 6, 2023Updated 3 years ago
- Library for 8-bit optimizers and quantization routines.☆777Aug 18, 2022Updated 4 years ago
- [EMNLP 2021] SimCSE: Simple Contrastive Learning of Sentence Embeddings https://arxiv.org/abs/2104.08821☆3,658Updated this week
- An Open-Source Package for Information Retrieval.☆441Oct 7, 2022Updated 4 years ago
- Binary Passage Retriever (BPR) - an efficient passage retriever for open-domain question answering☆176Jun 6, 2021Updated 5 years ago