Pinned Loading

1. RQ-VAE-Recommender RQ-VAE-Recommender Public

   [Pytorch] Generative retrieval model using semantic IDs from "Recommender Systems with Generative Retrieval"

   Python 819 114

2. RecTokens RecTokens Public

   [Pytorch] Efficient tokenization for recommendations and generative retrieval. Inspired by STATIC decoding from "Vectorizing the Trie"

   Python 7 1

3. Gaussian-Mixture-VAE Gaussian-Mixture-VAE Public

   [Pytorch] Minimal implementation of a Variational Autoencoder (VAE) with Categorical Latent variables inspired from "Categorical Reparameterization with Gumbel-Softmax".

   Python 7 2

4. rq-kmeans rq-kmeans Public

   [Pytorch, Triton] GPU-optimized implementation of kmeans algorithms and RQKmeans vector quantization.

   Python 7

5. zig_zag_sampling_dormancy zig_zag_sampling_dormancy Public

   C++ repo for the poster "Zig-zag sampling for genealogies with dormancy".

   C++

6. upriyam-cmu/EDGE-Rec upriyam-cmu/EDGE-Rec Public

   Python 3