Fast and Flexible Neural Click Models in JAX

The best ChatGPT that $100 can buy.

Flax is a neural network library for JAX that is designed for flexibility.

ICTIR 2025 - Unidentified and Confounded? Understanding Two-Tower Models for Unbiased Learning to Rank

An implementation of a deep learning recommendation model (DLRM)

An extremely fast Python package and project manager, written in Rust.

A dataloader, but for JAX

Simulating online bandits for ranking under position bias

A large-scale information-rich web dataset, featuring millions of real clicked query-document labels

SIGIR 2024 - Unbiased Learning to Rank Meets Reality: Lessons from Baidu's Large-Scale Search Dataset

Library for reading and processing ML training data.

A fast, simple and lightweight Bloom filter library for Python, implemented in Rust.

SIGIR 2024 - Train flax-based MonoBERT rankers on Baidu-ULTR

Codebase for RetroMAE and beyond.

Personal toolkit for bias estimation unbiased learning to rank

Sweeper plugin based on SMAC for Hydra.

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization

Simple, safe way to store and distribute tensors

All the handwritten notes 📝 and source code files 🖥️ used in my YouTube Videos on Machine Learning & Simulation (https://www.youtube.com/channel/UCh0P7KwJhuQ4vrzc3IRuw4Q)

Fish-like autosuggestions for zsh

A game theoretic approach to explain the output of any machine learning model.

Official repository of the paper "Efficient Neural Causal Discovery without Acyclicity Constraints"

Course notes for CS228: Probabilistic Graphical Models.

Acceptance rates for the major AI conferences

RecSys 2023 - Collaborative filtering algorithms are prone to mainstream-taste bias

The simplest, fastest repository for training/finetuning medium-sized GPTs.