Machine Learning Framework

PyTorch LightningPython

A PyTorch wrapper that automates engineering boilerplate for scalable AI model training and deployment.

pytorch-lightningPython

A deep learning framework to pretrain and finetune any AI model at any scale with zero code changes.

ChainerPython

A flexible Python deep learning framework using define-by-run dynamic computational graphs for neural network research.

DMTK - Microsoft Distributed Machine Learning Tookit

A toolkit for distributed machine learning featuring parameter server framework, topic modeling, gradient boosting, and word embedding.