Kris Wu's repositories
Awesome-Imitation-Learning
A curated list of awesome imitation learning resources and publications
dex-affordance
[CoRL 2022] Implementation of "Learning Generalizable Dexterous Manipulation from Human Grasp Affordance"
Elastic-DT
[NeurIPS 2023] Implementation of Elastic Decision Transformer
ai-generated-pokemon-rudalle
Python script to preprocess images of all Pokémon to finetune ruDALL-E
animegan2-pytorch
PyTorch implementation of AnimeGANv2
AnimeGANv2
[Open Source]. The improved version of AnimeGAN. Landscape photos/videos to anime
DALLE2-pytorch
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
deformable-ravens
Code for: https://berkeleyautomation.github.io/bags/
diffusion_policy
[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion
dmcontrol-generalization-benchmark
DMControl Generalization Benchmark
fc2-live-dl
Tool to download FC2 live streams
iCloud-Uploader
This script is an automated uploader to iCloud.
imagen-pytorch
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
mjrl
Reinforcement learning algorithms for MuJoCo tasks
PixivUtil2
Download images from Pixiv and more!
PokemonRedExperiments
Playing Pokemon Red with Reinforcement Learning
pytorch-soft-actor-critic
PyTorch implementation of soft actor critic
splat
WebGL 3D Gaussian Splat Viewer
stable-baselines3
PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
stylegan2-pytorch
Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement
TD3
Author's PyTorch implementation of TD3 for OpenAI gym tasks
TD3_BC
Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL
tixcraft_bot
MaxBot open source code bot
ucsddissertation
LaTeX class file for writing dissertations at UC San Diego
vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
vits
VITS: Conditional Variational Autoencoder with Adversarial Learning for End-to-End Text-to-Speech