Open Source Software

Mila’s mission is to be a global pole for scientific advances that inspires innovation and the progress of artificial intelligence (AI) for the benefit of all. As part of this mission, Mila recognizes the significant potential of AI and the importance of making research more open, interdisciplinary and accessible.

An Environment Centred on Open Science

Explore a selection of notable open source software efforts led by or co-developed with Mila researchers over the years.

Academic Torrent

Academic Torrents is scalable platform using BitTorrent which distributes the cost of hosting data to prevent the rise and fall of dataset hosting providers and the erasure of the data they host.

Arcade Learning Environment (ALE)

ALE is a reinforcement learning benchmark and a framework allowing researchers to develop AI agents for Atari 2600 games. It continues to be maintained by Mila researchers.

AxonDeepSeg

The AxonDeepSeg framework is a segmentation software for microscopy data of nerve fibers based on a convolutional neural network.

BabyAI

BabyAI is a testbed for training agents to understand and execute language commands.

Chester

Chester is a free and accessible prototype system that can be used by medical professionals to understand the reality of deep learning tools for chest X-ray diagnostics.

Distributed Evolutionary Algorithms in Python (DEAP)

Distributed Evolutionary Algorithms in Python (DEAP) is an evolutionary computation framework for rapid prototyping and testing of ideas.

Dopamine

A research framework for fast prototyping of reinforcement learning algorithms. Dopamine was co-developed by Professor Marc G. Bellemare at Google.

Enhance-FineGrained

Code and data for Contrasting Intra-Modal and Ranking Cross-Modal Hard Negatives to Enhance Visio-Linguistic Fine-grained Understanding.

Graphium

With rich and powerful built-in molecular featurizers, Graphium is an open-source library designed for graph representation learning on real-world chemistry tasks.

Hrepr

Hrepr outputs HTML/pretty representations for Python objects.

HyPyP

HyPyP is for Hyperscanning Python Pipeline for inter-brain connectivity analysis.

Ivadomed

Ivadomed is an integrated framework for medical image analysis with deep learning, based on PyTorch.

Jurigged

Jurigged lets you update your code while it runs.

MEDomics-UdeS

The MEDomics UdeS research laboratory, led by Professor Martin Vallières at the Université de Sherbrooke, has been focused on the creation of predictive models in health informatics since its founding in 2020.

MAPL

MAPL is a multimodal AI system capable of understanding images and text, while generating free-form text as output.

MilaBench

MilaBench is a repository of training benchmarks.

MiniGrid

Minimalistic Gridworld Environment (MiniGrid) gym is maintained by the Farama Foundation.

Myia

Myia (follow-up to Theano) is a differentiable programming language capable of supporting large scale high-performance computations (e.g. linear algebra) and their gradients.

Myriad

Myriad is a real-world testbed that aims to bridge the gap between trajectory optimization and deep learning. The environments and tools in Myriad can be used for trajectory optimization, system identification, imitation learning, and reinforcement learning.

Orion

A collaboration between Mila and IBM, Oríon is a black-box function optimization library with a key focus on usability and integrability for its users.

Paperoni

Paperoni allows users to search for scientific papers from the command line.

PyNM

PyNM is a lightweight python implementation of Normative Modeling making it approachable.

Ptera

Ptera allows you to instrument code from the outside by specifying a set of variables to watch in an arbitrary Python call graph and manipulate a stream of their values.

qMRLab

qMRLab is a MATLAB/Octave open-source software for quantitative MR image analysis. The main goal of the qMRLab project is to provide the community with software that makes data fitting, simulation and protocol optimization as easy as possible for a myriad of different quantitative models.

Sequoia

Sequoia is a software framework to unify continual learning research.

Shimming-Toolbox

Shimming-Toolbox is an open-source Python software package enabling a variety of MRI shimming (magnetic field homogenization) techniques.

SpeechBrain

SpeechBrain is an open-source, general-purpose PyTorch speech processing toolkit designed to make the research and development of neural speech processing technologies easier by being simple, flexible, user-friendly, and well-rounded.

Spinal Cord Toolbox

SCT is a comprehensive, free and open-source set of command-line tools dedicated to the processing and analysis of spinal cord MRI data.

StratiPy

Patients stratification with Graph-regularized Non-negative Matrix Factorization (GNMF) in Python.

TD3

Mila PhD student Scott Fujimoto, co-supervised by Doina Precup and David Meger, holds the open source code for TD3, one of the best performing current deep reinforcement learning methods.

Temporal Graph Benchmark

TGB provides an automated machine learning pipeline for reproducible and accessible temporal graph research, including data loading, experiment setup and performance evaluation.

TinyDiffRast

A tiny course that aims to explore, explain, and implement state-of-the-art approaches to differentiable triangle rasterization. The course covers a class of inverse rendering techniques that enable 3D scene parameters (materials, lights, cameras, geometry) to be learned from 2D image supervision.

Theano

Theano, one of the earliest programming frameworks for deep learning, originated at Mila and Université de Montréal. Theano is a Python library and optimizing compiler for manipulating and evaluating mathematical expressions. The development of Theano was completed in 2017.

TorchDrug

TorchDrug is an open-source machine learning platform for drug discovery, covering techniques ranging from graph machine learning, deep generative models to reinforcement learning.

TorchProtein

Available as part of TorchDrug, TorchProtein is a ML library for protein science, providing representation learning models for both protein sequences and structures, as well as fundamental protein tasks like function prediction and structure prediction.