close-up image of interlocking gears turning with a rainbow gradient overlay

Research Tools: code, datasets, & models

Discover an index of datasets, SDKs, APIs and open-source tools developed by Microsoft researchers and shared with the global academic community below. These experimental technologies—available through Azure AI Foundry Labs (opens in new tab)—offer a glimpse into the future of AI innovation.

Current selections

Sort by: Most recent

Clear selections

Search within these results

Published Date

Dataset Source Code

Tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation. Supported Framework: Pytorch Supported GPUs: CUDA(fp32 + fp16), ROCm(fp32 + fp16)

GitHub Project

Dataset Source Code

FOST

FOST(Forecasting open source tool) aims to provide an easy-use tool for spatial-temporal forecasting. The users only need to organize their data into a certain format and then get the prediction results with one command. FOST…

GitHub Video

Dataset Source Code

DiCE: A library for generating Diverse Counterfactual Explanations

DiCE is a Python library that can generate counterfactual explanations for any machine learning classifier. Counterfactual explanations present “what-if” perturbations of the input such that an ML classifier outputs a different class for those perturbations…

GitHub Project Project

Tool

Odia Speech Data and Model

As part of this release, Navana Tech and Microsoft Research India are open-sourcing 1648 hours of validated Odia speech dataset and a baseline model for Odia speech recognition. The speech dataset consists of recordings in…

Access Project

Dataset Source Code

LiST (Lite Self-Training)

We present a new method LiST for efficient fine-tuning of large pre-trained language models (PLMs) in few-shot learning settings. LiST significantly improves over recent methods that adopt prompt fine-tuning using two key techniques. The first…

GitHub Project Publication

Dataset Source Code

LITMUS Predictor

LITMUS Predictor provides support for simulating performance in ~100 languages given training observations of the desired task-model. Each training observation specifies the finetuning-datasize + test-performance in different languages. Further, the tool provides support for constructing…

GitHub Project Publication Publication

Dataset Source Code

DistIR

DistIR is an intermediate representation (IR) and associated set of tools for optimizing distributed machine learning computations (both training and inference). An IR is a format for representing programs used by compilers and software analysis…

GitHub Publication

Dataset Source Code

Modulo Paper repository

Modulo allows optimal selection of vehicles for effective drive-by sensing.

GitHub Project Publication

Dataset Source Code

Machine Intelligence PyTorch Module Zoo

A number of PyTorch modules (neural network components) that are broadly (re)usable. The goal of this release is to provide a library to machine learning researchers to use and advance the state-of-the-art in the area.

GitHub

Dataset Source Code

Stochastic Mixture-of-Experts

This PyTorch package implements Taming Sparsely Activated Transformer with Stochastic Experts.

GitHub