Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

A toolkit for developing and comparing reinforcement learning algorithms.

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

Deep Learning for humans

Open Source Computer Vision Library

scikit-learn: machine learning in Python

The agent engineering platform

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

An Open Source Machine Learning Framework for Everyone

Tensors and Dynamic neural networks in Python with strong GPU acceleration