An open-source machine learning framework that accelerates the path from research prototyping to production deployment.

Keras is a deep learning API written in Python, running on top of the machine learning platform TensorFlow.

TensorFlow is an end-to-end open-source platform for machine learning. It has a comprehensive, flexible ecosystem of tools, libraries, and community resources that lets researchers push the state-of-the-art in ML and developers easily build and deploy ML-powered applications.

JAX is Autograd and XLA, brought together for high-performance machine learning research. With its updated version of Autograd, JAX can automatically differentiate native Python and NumPy functions. It can differentiate through loops, branches, recursion, and closures, and it can take derivatives of derivatives. It supports reverse-mode differentiation (a.k.a. backpropagation) via grad as well as forward-mode differentiation, and the two can be composed arbitrarily to any order.

PyTorch Lightning is a high level framework to abstract away your PyTorch boilerplate code

Hugging Face Transformers provides APIs to easily download and train state-of-the-art pre-trained models. With Transformers you can train Text, Image, Audio, and Multi-Modal models

YOLOv5 ? is a family of object detection architectures and models pretrained on the COCO dataset.

spaCy is a free, open-source library for advanced Natural Language Processing (NLP) in Python. spaCy is designed specifically for production use and helps you build applications that process and “understand” large volumes of text. It can be used to build information extraction or natural language understanding systems, or to pre-process text for deep learning.

XGBoost is an optimized distributed gradient boosting library designed to be highly efficient, flexible and portable.

LightGBM is a gradient boosting framework that uses tree based learning algorithms.

MMDetection is an open source object detection toolbox based on PyTorch. It is a part of the OpenMMLab project.

scikit-learn is a Python library for simple and efficient tools for predictive data analysis, built on NumPy, SciPy, and matplotlib

Fastai is a deep learning library which provides practitioners with high-level components that can quickly and easily provide state-of-the-art results in standard deep learning domains, and provides researchers with low-level components that can be mixed and matched to build new approaches.

TensorLy provides all the utilities to easily use tensor methods, from core tensor operations and tensor algebra to tensor decomposition and regression. Write your code once and execute in using any of the supported frameworks, enabling tensor learning on GPU, multi-machines, and deep tensorized learning

ZenML is an extensible, open-source MLOps framework to create production-ready machine learning pipelines. Built for data scientists, it has a simple, flexible syntax, is cloud- and tool-agnostic, and has interfaces/abstractions that are catered towards ML workflows. At its core, ZenML pipelines execute ML-specific workflows from sourcing data to splitting, preprocessing, training, all the way to the evaluation of results and even serving. There are many built-in batteries to support common ML development tasks. ZenML is not here to replace the great tools that solve these individual problems. Rather, it integrates natively with popular ML tooling and gives standard abstraction to write your workflows.

TorchDrug is a graph representation learning toolbox built on top of the PyTorch platform.

OpenAI’s API gives practitioners access to GPT-3, an incredibly powerful natural language model that can be applied to virtually any task that involves understanding or generating natural language.

Catalyst is a PyTorch framework for deep learning R&D that focuses on reproducibility, rapid experimentation, and codebase reuse so you can create something new.

Test Suites for Validating ML Models & Data

Hydra is an open-source Python framework that simplifies the development of research and other complex applications.

Kaggle is an online community of data scientists and machine learning practitioners that allows users to find and publish data sets, explore and build models in a web-based data-science environment, work with other data scientists and machine learning engineers, and enter competitions to solve data science challenges.

PyTorch Geometric is a library built upon PyTorch to easily write and train Graph Neural Networks (GNNs) for a wide range of applications related to structured data.