Official Code Base for ICLR 2024 paper Enhancing Tail Performance in Extreme Classifiers by Label Variance Reduction

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

Dual-encoders for Extreme Multi-label Learning

get things from one computer to another, safely

Repository to prototype ML ideas and take notes on ML concepts.

An open platform for training, serving, and evaluating large language models. Release repo for Vicuna and Chatbot Arena.

Official codebase for NeurIPS 2022 paper End-to-end Learning to Index and Search in Large Output Spaces

Multimodal extreme classification

VL-LTR: Learning Class-wise Visual-Linguistic Representation for Long-Tailed Visual Recognition

This is the official codebase for KDD 2021 paper Generalized Zero-Shot Extreme Multi-Label Learning

Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch

PECOS - Prediction for Enormous and Correlated Spaces

BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)

Tools for multi-label classification problems.

VIP cheatsheets for Stanford's CS 229 Machine Learning

Algorithms