TextGAN is a PyTorch framework for Generative Adversarial Networks (GANs) based text generation models.

🦉 OWL: Optimized Workforce Learning for General Multi-Agent Assistance in Real-World Task Automation

A Flexible Framework for Experiencing Cutting-edge LLM Inference Optimizations

The collection of papers about Private Evolution

Is synthetic data from generative models ready for image recognition?

We are building an open database of COVID-19 cases with chest X-ray or CT images.

FL testbed

[AAAI'22] HarmoFL: Harmonizing Local and Global Drifts in Federated Learning on Heterogeneous Medical Images

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.

PyTorch code and models for the DINOv2 self-supervised learning method.

A curated list of resources for using LLMs to develop more competitive grant applications.

This is the implementation for Web Search & Mining course, which contains data crawling (Ubuntu IRC data (2004~2021)) and preprocessing, boolean search, spell-correction with tolerant (fuzzy) searc…

GLIDE: a diffusion-based text-conditional image synthesis model

Code for visualizing the loss landscape of neural nets

[EMNLP 2022] Code for our paper “ZeroGen: Efficient Zero-shot Learning via Dataset Generation”.

Private Evolution: Generating DP Synthetic Data without Training [ICLR 2024, ICML 2024 Spotlight]

Model, Code & Data for the EMNLP'23 paper "Making Large Language Models Better Data Creators"

A unified framework for privacy-preserving data analysis and machine learning

Inference code for CodeLlama models

Large Language Model (LLM) Systems Paper List

Federated Learning Systems Paper List

[DMLR 2024] FedAIoT: A Federated Learning Benchmark for Artificial Intelligence of Things

higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual training steps.

[NeurIPS 2023] Learning Transformer Programs

中文语言理解测评基准 Chinese Language Understanding Evaluation Benchmark: datasets, baselines, pre-trained models, corpus and leaderboard

Inference code for Llama models

A basic 2D maze environment where an agent start from the top left corner and try to find its way to the bottom left corner.

very easy implementation of dueling DQN in pytorch

A toolkit for developing and comparing reinforcement learning algorithms.