TensorFlowtensorflow/tensorflow
Open source software library for numerical computation using data flow graphs.
609 libraries, 201 updated this year
The broadest part of the list: libraries that train and run models without being tied to one kind of data. It mixes two generations. Classical toolkits such as scikit-learn, XGBoost, LightGBM and CatBoost fit models to tables of features and remain the first choice for structured business data. Deep learning frameworks such as TensorFlow, PyTorch, JAX and Keras build neural networks of any shape and run them on GPUs.
609 libraries
Open source software library for numerical computation using data flow graphs.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
OpenCV has C++, C, Python, Java and MATLAB interfaces and supports Windows, Linux, Android and Mac OS.
A Python module for machine learning built on top of SciPy.
High-level neural networks frontend for TensorFlow, CNTK and Theano.
Streamlit is an framework to create beautiful data apps in hours, not weeks.
A Python library for quickly creating and sharing demos of models. Debug models interactively in your browser, get feedback from collaborators, and generate public links without deploying anything.
An open-source deep learning system for large-scale model training and inference with high efficiency and low cost.
Open Source framework to streamline use of neural networks.
JAX is Autograd and XLA, brought together for high-performance machine learning research.
A deep learning framework developed with cleanliness, readability, and speed in mind.
Implementations of Machine Learning models from scratch in Python with a focus on transparency. Aims to showcase the nuts and bolts of ML in an accessible way.
The lightweight PyTorch wrapper for high-performance AI research.
High-level wrapper built on the top of Pytorch which supports vision, text, tabular data and collaborative filtering.
Book/iPython notebooks on Probabilistic Programming in Python.
Darknet is an open source neural network framework written in C and CUDA. It is fast, easy to install, and supports CPU and GPU computation.
> Graph Neural Network Library for PyTorch.
Open source engineering platform to debug, evaluate, and monitor your LLM applications, RAG systems, and agentic workflows with comprehensive tracing, automated evaluations, and production-ready dashboards. (Source Code)
Examples and best practices for building recommendation systems, provided as Jupyter notebooks. The repo contains some of the latest state of the art algorithms from Microsoft Research as well as from other companies and institutions.
Candle is a minimalist ML framework for Rust with a focus on performance (including GPU support) and ease of use.
Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Go, JavaScript and more.
A WebGL accelerated, browser based JavaScript library for training and deploying ML models.
Microsoft's fast, distributed, high performance gradient boosting (GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.
Code samples for my book "Neural Networks and Deep Learning".
The Computational Network Toolkit (CNTK) by Microsoft Research, is a unified deep-learning toolkit that describes neural networks as a series of computational steps via a directed graph.
A relational column-oriented database designed for real-time analytics on time series and event data.
Reference implementations of ML models written in numpy
Optuna is an automatic hyperparameter optimization software framework, particularly designed for machine learning.
Neural networks in JavaScript - continued community fork of Brain.
DLib has C++ and Python interfaces for face detection and training general object detectors.
Uncover insights, surface problems, monitor and fine tune your generative LLM, CV and tabular models.
The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.
ConvNetJS is a JavaScript library for training Deep Learning models
Machine learning from Apple. Turi Create simplifies the development of custom machine learning models. You don't have to be a machine learning expert to add recommendations, object detection, image classification, image similarity or activity classification to your app.
AutoML for Image, Text, Tabular, Time-Series, and MultiModal Data.
> Python Outlier Detection, comprehensive and scalable Python toolkit for detecting outlying objects in multivariate data. Featured for Advanced models, including Neural Networks/Deep Learning and Outlier Ensembles.
Tool that automatically creates and optimizes machine learning pipelines using genetic programming. Consider it your personal data science assistant, automating a tedious part of machine learning.
A Python library for secure and private Deep Learning built on PyTorch and TensorFlow.
Optimizing GPU-meta-programming code generating array oriented optimizing math compiler in Python.
An open-source, low-code machine learning library in Python that automates machine learning workflows.
Deep learning operations reinvented (for pytorch, tensorflow, jax and others).
Deep learning library featuring a higher-level API for TensorFlow.