zhangjb416's starred repositories
factor-world
Decomposing the Generalization Gap in Imitation Learning for Visual Robotic Manipulation (2023)
flash-attention
Fast and memory-efficient exact attention
pytorch-open-x-embodiment
Data pre-processing and training code on Open-X-Embodiment with pytorch
RoboFlamingo
Code for RoboFlamingo
ManiSkill-Learn
ManiSkill-Learn is a framework for training agents on SAPIEN Open-Source Manipulation Skill Challenge (ManiSkill Challenge), a physics-rich manipulation skill benchmark with large-scale demonstrations.