[D] Is there an open-source implementation of the Retrieval-Enhanced Transformer (RETRO)?

This page summarizes the projects mentioned and recommended in the original post on /r/MachineLearning

Our great sponsors
  • WorkOS - The modern identity platform for B2B SaaS
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • SaaSHub - Software Alternatives and Reviews
  • deepmind-research

    This repository contains implementations and illustrative code to accompany DeepMind publications

    checked their repository and indeed it isn't open sourced yet https://github.com/deepmind/deepmind-research i wouldn't mind taking this on. the only novel thing is the chunked-cross attention (maintaining causality). does your team / company sponsor open-source efforts?

  • RETRO-pytorch

    Implementation of RETRO, Deepmind's Retrieval based Attention net, in Pytorch

    i'll give it a shot https://github.com/lucidrains/RETRO-pytorch 👍

  • WorkOS

    The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  • faiss

    A library for efficient similarity search and clustering of dense vectors.

    Faisss now includes the 4bit PQ algorithm from scann (https://github.com/facebookresearch/faiss/issues/1399). Looks to me like faiss better maintained and has other options but I don't have any specific experience with these.

  • TorchPQ

    Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda

    if you're decided to run nearest neighbor search on GPU, you might wanna give TorchPQTorchPQ a try

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts