Research Tools: code, datasets, & models

MOFDiff is a diffusion model for generating coarse-grained MOF structures. This codebase also contains the code for deconstructing/reconstructing the all-atom MOF structures to train MOFDiff and assemble CG structures generated by MOFDiff.

The AI Controller Interface is a system design and implementation that enables customer user code (AI Controllers, implemented as light-weight virtual machines) to tightly, efficiently, and securely integrate with LLM decoding in a cloud service.…

UDOP adopts an encoder-decoder Transformer architecture based on T5 for document AI tasks like document image classification, document parsing and document visual question answering. You can use the model for document image classification, document parsing…

Node Engine is a Python service that executes a computational flow. It is designed for rapid prototyping of services and applications, e.g. used as a chatbot service in a larger system. Each call to the service…

Diffy is a research prototype tool that analyzes JSON configuration files. The goal of Diffy is to assist with configuration management. It compares JSON files and warns about potential issues by finding configuration settings that…

Official code release of our EMNLP 2023 work NeuralInvariantRanker (opens in new tab). We have designed a ranker that can distinguish between correct inductive invariants and incorrect attempts based on the problem definition. The ranker…

🕮 KITAB is a challenging dataset and a dynamic data collection approach for testing abilities of Large Language Models (LLMs) in answering information retrieval queries with constraint filters. A filtering query with constraints can be…

Structure-aware Multi-Objective Metaprompt Optimization Library for Python. A flexible, easy-to-use library for running and optimizing prompts for Large Language Models (LLMs).

CodePlan is a research project that formalizes repository-level coding tasks as planning problems and uses static analysis and large language models (LLMs) to solve them. This replication package is for the paper titled “CodePlan: Repository-level…

Algorithms to Handle Signal Delay in Deep Reinforcement Learning aims to address the problem of signal delay in continuous robotic control. Signal delay occurs when there is a lag between an agent’s perception of the…