Research Tools: code, datasets, & models

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.…

Research code for CVPR 2021 paper “End-to-End Human Pose and Mesh Reconstruction with Transformers”

We release code and datasets to train FLIN, a natural language model for web navigation.

This project is a large dataset, along with baseline trained machine learning models, for the tasks of table detection and table structure recognition in scientific PDF documents. Current datasets for table structure recognition are small…

The library implements the parallel collision search algorithm by van Oorschot and Wiener (vOW) on hardware, with the goal of concretely assessing the classical security of the Supersingular Isogeny Key Encapsulation (SIKE) protocol. SIKE is…