Stars
Official Code of Memento: Fine-tuning LLM Agents without Fine-tuning LLMs
Minimal, lightweight JAX implementations of popular models.
An open-source AI agent that brings the power of Gemini directly into your terminal.
Official PyTorch implementation of "GuidedQuant: Large Language Model Quantization via Exploiting End Loss Guidance" (ICML 2025)
Doing simple retrieval from LLM models at various context lengths to measure accuracy
Redirects the user to a random URL from a list of URLs
Fast and memory-efficient exact attention
A Python toolbox for performing gradient-free optimization
[COLM 2024] LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Composition
[NeurIPS 2022] A Fast Post-Training Pruning Framework for Transformers
Scenic: A Jax Library for Computer Vision Research and Beyond
Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".
[NeurIPS'22] Tokenized Graph Transformer (TokenGT), in PyTorch
BASQ: Branch-wise Activation-clipping Search Quantization for Sub-4-bit Neural Networks, ECCV 2022
Bio-Computing Platform Featuring Large-Scale Representation Learning and Multi-Task Deep Learning “螺旋桨”生物计算工具集
[CVPR 2022 Oral, Best Paper Finalist] Official PyTorch implementation of FIFO
EMNLP 2021 - Pre-training architectures for dense retrieval
Official Implementation of "Domain Specific Batch Normalization for Unsupervised Domain Adaptation (CVPR2019)"
A Python/Pytorch app for easily synthesising human voices
StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/unconditional image generation.
SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]
Chemical-Reaction-Aware Molecule Representation Learning
Official Code Repository for the paper "Edge Representation Learning with Hypergraphs" (NeurIPS 2021)
Graphormer is a general-purpose deep learning backbone for molecular modeling.