Awesome Machine Learning

An Open Source Machine Learning Framework for Everyone

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Open Source Computer Vision Library

Ultralytics YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

Milvus is a high-performance, cloud-native vector database built for scalable vector ANN search

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

Caffe: a fast open framework for deep learning.

OpenPose: Real-time multi-person keypoint detection library for body, face, hands, and foot estimation

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

MLX: An array framework for Apple silicon

Graph Neural Network Library for PyTorch

Apache Hadoop

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

Vowpal Wabbit is a machine learning system which pushes the frontier of machine learning with techniques such as online, hashing, allreduce, reductions, learning2search, active, and interactive learning.

Out-of-Core hybrid Apache Arrow/NumPy DataFrame for Python, ML, visualization and exploration of big tabular data at a billion rows per second 🚀

Rust bindings for the C++ api of PyTorch.

An Engine-Agnostic Deep Learning Framework in Java

The AI-native database built for LLM applications, providing incredibly fast hybrid search of dense vector, sparse vector, tensor (multi-vector), and full-text.

High performance, easy-to-use, and scalable machine learning (ML) package, including linear model (LR), factorization machines (FM), and field-aware factorization machines (FFM) for Python and CLI interface.

Shōgun

ThunderSVM: A Fast SVM Library on GPUs and CPUs

ThunderGBM: Fast GBDTs and Random Forests on GPUs