Research Tools: code, datasets, & models

We view Large Language Models as stochastic language layers in a network, where the learnable parameters are the natural language prompts at each layer. We stack two such layers, feeding the output of one layer…

In the Harnessing AutoMobiles for Safety, or HAMS, project, we use low-cost sensing devices to construct a virtual harness for vehicles. The goal is to monitor the state of the driver and how the vehicle…

Code and Data artifact for NeurIPS 2023 paper – “Monitor-Guided Decoding of Code LMs with Static Analysis of Repository Context”. `multispy` is a lsp client library in Python intended to be used to build applications…

Official PyTorch implementation for the paper: “Renee: End-to-end training of extreme classification models” accepted at MLSys 2023.

INTREPID (stands for INTeractive learning via REPresentatIon Discovery) is a library that contains various interactive learning algorithms that learn a representation (or a latent state) from observational data in order to complete their tasks.

Chart Reader is a web-based accessibility engine, which enables rendering of accessible visualizations for screen reader uses to read and better understand the visualizations and underlying data.

This repository contains the official code for our IEEE S&P 2023 paper using GPT-2 language models and Flair Named Entity Recognition (NER) models. It allows fine-tuning (i) undefended, (ii) differentially-private and (iii) scrubbed language models…

Syntheseus is a package for retrosynthetic planning. It contains implementations of common search algorithms and a simple API to wrap custom reaction models and write custom algorithms. It is meant to allow for simple benchmarking…

Guidance enables you to control modern language models more effectively and efficiently than traditional prompting or chaining. Guidance programs allow you to interleave generation, prompting, and logical control into a single continuous flow matching how…

HI-ML toolbox for deep learning for medical imaging and Azure integration. The Microsoft Health Intelligence Machine Learning Toolbox aims at providing low-level and high-level building blocks for Machine Learning / AI researchers and practitioners. It…