Research Tools: code, datasets, & models

Varuna is a tool for efficient training of large DNN models on commodity GPUs and networking. It implements a combination of pipeline parallelism and data parallelism in PyTorch, and enables training on a changing set…

Interactive Machine Translation (INMT) interface assists human translators with on-the-fly hints and suggestions. This makes the end-to-end translation process faster, more efficient, and creates high-quality translations. We augment the OpenNMT backend with a mechanism to…

This is the official code repository to accompany the paper CausalCity: Complex Simulations with Agency for Causal Discovery and Reasoning. Here we provide Python code for generating and logging scenarios using the simulation environment as…

This library is a collection of GitHub repositories containing software applications, data sets, and hardware reference designs to support research and education in the fields of service-robotics. The term service-robotics refers to the expectation that the robots can…

Synthesized Collective Communication Library (SCCL) is an algorithm synthesizer for GPU communication in machine learning workloads. SCCL-Runtime, which is what we are seeking the release approval for, takes a synthesized algorithm from SCCL and runs…

Koka: a Functional Language with Effects Koka is a strongly typed functional-style language with effect types and handlers.

This repository showcases building task-oriented bot at scale with handful examples via fine-tuning a pretrained model using SOLOIST framework, and contains the dataset, source code and pre-trained model for the following paper: SOLOIST: Building Task…

Official PyTorch code for ClipBERT, an efficient framework for end-to-end learning for image-text and video-text tasks. It takes raw videos/images + text as inputs, and outputs task predictions. ClipBERT is designed based on 2D CNNs…

We have developed a new algorithm to optimally compose privacy guarantees of differentially private (DP) algorithms to arbitrary accuracy. This release accompanies our research paper. We hope to help the privacy community run more accurate…

Analyses of text corpora over time can reveal trends in beliefs, interest, and sentiment about a topic. We focus on views expressed about artificial intelligence (AI) in the New York Times over a 30-year period.…