Yuhuai(Tony) Wu

eval_gen. Evaluation code with models for the paper "On the Quantitative Analysis of Decoder-Based Generative Models"

LIME. Official code for paper LIME: Learning Inductive Bias for Primitives of Mathematical Reasoning

python-ggp. General Game Playing (GGP) framework in Python

adversarial. Code and hyperparameters for the paper "Generative Adversarial Networks"

gmmn. Generative moment matching networks

ray. A system for parallel and distributed Python that unifies the ML ecosystem.

flax. Flax is a neural network ecosystem for JAX that is designed for flexibility.

LiveCodeBench. Official repository for the paper "LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Code"

trax. Trax — Deep Learning with Clear Code and Speed

pytorch-a2c-ppo-acktr-gail. PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

jax-triton. jax-triton contains integrations between JAX and OpenAI Triton