Building Recommendation Systems in Python and JAX — a fork migrated from TensorFlow/JAX/Flax to pure PyTorch (Python 3.11+). Three sub-projects: Pinterest Shop-the-Look (two-tower CNN, triplet loss), Spotify Million Playlist (embedding-based retrieval, contrastive learning), Wikipedia NLP (GloVe co-occurrence + LSTM text-to-URL).
python nlp spotify machine-learning deep-learning pinterest pytorch collaborative-filtering pyspark recommendation-system recommender-system glove embedding playlist-continuation wandb two-tower-model wikipedia-nlp
-
Updated
Jun 9, 2026 - TeX