Open standard for machine learning interoperability

A unified framework for machine learning with time series

Official repo for consistency models.

Merlion: A Machine Learning Framework for Time Series Intelligence

Compute FID scores with PyTorch.

[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion

Variational autoencoder implemented in tensorflow and pytorch (including inverse autoregressive flow)

Library for detecting anomalies in signals

Official implementation of the paper "Towards Faster and Stabilized GAN Training for High-fidelity Few-shot Image Synthesis" in ICLR 2021

Evolution Strategies in PyTorch

Image Captioning Using Transformer

UMI on Legs: Making Manipulation Policies Mobile with Manipulation-Centric Whole-body Controllers

Sample code for Constrained Graph Variational Autoencoders

Data augmentation using synthetic data for time series classification with deep residual networks

[CoRL 2022] Deep Whole-Body Control: Learning a Unified Policy for Manipulation and Locomotion

Temporal Fusion Transformers for Tensorflow 2.x

Train a loco-manipulation dog with RL

[RSS 2024] Consistency Policy: Accelerated Visuomotor Policies via Consistency Distillation

TensorFlow implementation for the GP-VAE model described in https://arxiv.org/abs/1907.04155

Code for "Multi-Time Attention Networks for Irregularly Sampled Time Series", ICLR 2021.

(NeurIPS 2020 Spotlight) Wasserstein Distances for Stereo Disparity Estimation

Using LSTM or Transformer to solve Image Captioning in Pytorch

Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation

Code to train a compliant whole-body controller for the Unitree B1 + Z1

TensorFlow implementation of DeepTCN model for probabilistic time series forecasting with temporal convolutional networks.

Code for "Diffusion Co-Policy for Synergistic Human-Robot Collaborative Tasks", 2023.