Stars
GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
UFACTORY model support for RL environment simulation, based on panda-gym framework
An example RLDS dataset builder for X-embodiment dataset conversion.
Personal repository for quick RL prototyping. Work in progress!
A PyTorch implementation of the CVPR 2022 Paper "Neural RGB-D Surface Reconstruction"
Implementation of MusicLM, Google's new SOTA model for music generation using attention networks, in Pytorch
Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Research, in Pytorch