Open Source Software

Mila > Open Source Software

A high-performing open science environment

Mila’s mission is to be a global pole for scientific advances that inspires innovation and the progress of 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.

Open Source Software

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

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.

Discover Theano

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. 

Discover Myia

MiniGrid

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

Discover MiniGrid

BabyAI


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

Discover BabyAI

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.

Discover SpeechBrain

Distributed Evolutionary Algorithms in Python (DEAP)

DEAP is a novel evolutionary computation framework for rapid prototyping and testing of ideas, seeking to make algorithms explicit and data structures transparent.

Discover DEAP

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.

Discover TorchDrug

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.

Discover TorchProtein

Oríon


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. 

Discover Oríon

ivadomed

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

Discover ivadomed

Dopamine

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

Discover Dopamine

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.

Discover ALE

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.

Discover TD3

AxonDeepSeg


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

Discover AxonDeepSeg

MilaBench

MilaBench is a repository of training benchmarks. 

Discover MilaBench

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.

Discover Ptera

Sequoia

A software framework to unify continual learning research

Discover Sequoia 

Paperoni

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

Discover Paperoni

Jurriged

Jurigged lets you update your code while it runs.

Discover Jurriged

Hrepr

HTML representation for Python objects.

Discover Hrepr

Academic Torrents


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.

Discover Academic Torrents

Chester, the AI Radiology Assistant


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.

Discover Chester