Train Deep Learning Models Tools

Chainer

Development

The pioneer of dynamic computational graphs for high-performance deep learning and research.

Apache MXNet

Development

The high-performance deep learning framework for flexible and efficient distributed training.

Deeplearning4j

Development

A suite of tools for deploying and training deep learning models using the JVM.

Fairseq

Development

The high-performance sequence modeling toolkit for researchers and production-grade NLP engineering.

Catalyst

Development

PyTorch framework for Deep Learning R&D focusing on reproducibility and rapid experimentation.

MMDetection

Computer Vision

The industry-standard open-source object detection toolbox for academic research and industrial deployment.

Tensor2Tensor

General AI

A library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

Places365

Computer Vision

The industry-standard deep learning dataset and model suite for state-of-the-art scene recognition.

Google Colaboratory

Machine Learning Platform

A cloud-based Jupyter notebook environment for rapid AI development with seamless GPU/TPU access.

Keras

Machine Learning Framework

The high-level deep learning API for JAX, PyTorch, and TensorFlow.