Eddie Zhang's repositories
accelerate
🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision
calvin
CALVIN - A benchmark for Language-Conditioned Policy Learning for Long-Horizon Robot Manipulation Tasks
challenge-iclr-2022
GitHub repository for the ICLR Computational Geometry & Topology Challenge 2021
ColossalAI
Colossal-AI: A Unified Deep Learning System for Big Model Era
CQL
Conservative Q Learning on top of SAC
d4rl-installable
A benchmark for offline reinforcement learning.
diffusers-groups
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
gym
A toolkit for developing and comparing reinforcement learning algorithms.
ibex
Ibex is a small 32 bit RISC-V CPU core, previously known as zero-riscy.
imagen-pytorch
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
learning
all my repos that are purely for experimentation
LegoFormer
Attention-based block-by-block 3D object reconstruction using Transformer
MWP-BERT
NAACL 2022 Findings Paper: MWP-BERT: Numeracy-Augmented Pre-training for Math Word Problem Solving
PyRTL
A collection of classes providing simple hardware specification, simulation, tracing, and testing suitable for teaching and research. Simplicity, usability, clarity, and extendability rather than performance or optimization is the overarching goal.
stablediffusion
High-Resolution Image Synthesis with Latent Diffusion Models