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.

CAI Neural APIPascal

A Pascal-based deep learning neural network API optimized for AVX/AVX2/AVX512 and OpenCL, supporting AMD, Intel, and NVIDIA hardware.

skproPython

A scikit-learn compatible Python library for probabilistic regression, survival analysis, and probability distributions.