Deep learning libraries

251 libraries, 105 updated this year

Libraries for building and training neural networks. The center of gravity is PyTorch, TensorFlow and JAX, with Keras as a high-level layer that now runs on all three. Around them sit model libraries such as Hugging Face Transformers for language and timm for images, and training helpers like PyTorch Lightning and fastai that remove boilerplate from the training loop.

251 libraries

TensorFlowtensorflow/tensorflow

Open source software library for numerical computation using data flow graphs.

PythonGeneral

201.2k updated today

Transformershuggingface/transformers

A deep learning library containing thousands of pre-trained models on different tasks. The goto place for anything related to Large Language Models.

PythonLLMsNLPSpeech

168.1k updated today

PyTorchpytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PythonGeneral

105k updated today

OpenCVopencv/opencv

OpenCV has C++, C, Python, Java and MATLAB interfaces and supports Windows, Linux, Android and Mac OS.

C++GeneralVision

91.3k updated today

keraskeras-team/keras

High-level neural networks frontend for TensorFlow, CNTK and Theano.

PythonGeneral

64.4k updated 1d ago

YOLOv8ultralytics/ultralytics

Ultralytics' YOLOv8 implementation with C++ support for real-time object detection and tracking, optimized for edge devices.

CVision

62.8k updated today

Streamlitstreamlit/streamlit

Streamlit is an framework to create beautiful data apps in hours, not weeks.

PythonGeneralData & viz

46.1k updated today

RLlibray-project/ray

RLlib is an industry level, highly scalable RL library for tf and torch, based on Ray. It's used by companies like Amazon and Microsoft to solve real-world decision making problems at scale.

PythonRL

44.1k updated today

Colossal-AIhpcaitech/ColossalAI

An open-source deep learning system for large-scale model training and inference with high efficiency and low cost.

PythonGeneral

41.4k updated 2d ago

MindsDBmindsdb/mindsdb

Open Source framework to streamline use of neural networks.

PythonGeneral

39.8k updated 1d ago

timmrwightman/pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more.

PythonVision

37.2k updated 4d ago

JAXgoogle/jax

JAX is Autograd and XLA, brought together for high-performance machine learning research.

PythonGeneral

36.5k updated today

Qdrantqdrant/qdrant

Qdrant is open source vector similarity search engine with extended filtering support, written in Rust.

LLMsMLOps

35.2k updated 1d ago

detectron2facebookresearch/detectron2

FAIR's next-generation research platform for object detection and segmentation. It is a ground-up rewrite of the previous version, Detectron, and is powered by the PyTorch deep learning framework.

PythonVision

34.9k updated 4d ago

CaffeBVLC/caffe

A deep learning framework developed with cleanliness, readability, and speed in mind.

General

34.6k updated 2y ago

Netronlutzroeder/netron

Visualizer for machine learning models.

JavaScriptGeneralMLOps

33.6k updated today

ML-From-Scratcheriklindernoren/ML-From-Scratch

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.

PythonGeneralRL

33.1k updated 3y ago

PyTorch LightningPyTorchLightning/pytorch-lightning

The lightweight PyTorch wrapper for high-performance AI research.

PythonGeneral

31.4k updated 12d ago

Fastaifastai/fastai

High-level wrapper built on the top of Pytorch which supports vision, text, tabular data and collaborative filtering.

PythonGeneral

28.2k updated 1mo ago

Darknetpjreddie/darknet

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.

CGeneral

26.5k updated 2y ago

Detectronfacebookresearch/Detectron Archived

FAIR's software system that implements state-of-the-art object detection algorithms, including Mask R-CNN. It is written in Python and powered by the Caffe2 deep learning framework.

PythonVision

26.3k archived

PyTorch Geometricpyg-team/pytorch_geometric

> Graph Neural Network Library for PyTorch.

PythonGeneral

24.1k updated 2d ago

Microsoft RecommendersMicrosoft/Recommenders

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.

PythonGeneral

22k updated 3d ago

Jina AIjina-ai/serve

An easier way to build neural search in the cloud. Compatible with Jupyter Notebooks.

Python

21.9k updated 1y ago

MXNetapache/incubator-mxnet Archived

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Go, JavaScript and more.

+3General

20.8k archived

TensorFlow.jstensorflow/tfjs

A WebGL accelerated, browser based JavaScript library for training and deploying ML models.

JavaScriptGeneral

19.2k updated 4mo ago

Neural Networks and Deep Learningmnielsen/neural-networks-and-deep-learning

Code samples for my book "Neural Networks and Deep Learning".

PythonGeneral

18k updated 2y ago

CNTKMicrosoft/CNTK Archived

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.

General

17.6k archived

numpy-MLddbourgin/numpy-ml

Reference implementations of ML models written in numpy

PythonGeneralNLP

16.3k updated 2y ago

burntracel-ai/burn

Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals

Rust

16.2k updated today

OpenFacecmusatyalab/openface

Free and open source face recognition with deep neural networks.

PythonVision

15.4k updated 10d ago

albumentationsalbu/albumentations Archived

А fast and framework agnostic image augmentation library that implements a diverse set of augmentation techniques. Supports classification, segmentation, detection out of the box. Was used to win a number of Deep Learning competitions at Kaggle, Topcoder and those that were a part of the CVPR workshops.

PythonVision

15.3k archived

Voskalphacep/vosk-api

An offline speech recognition toolkit with C++ support, designed for low-resource devices and multiple languages.

C++Speech

15.2k updated 9d ago

Brain.jsBrainJS/brain.js

Neural networks in JavaScript - continued community fork of Brain.

JavaScriptGeneral

14.9k updated 2y ago

DLibdavisking/dlib

DLib has C++ and Python interfaces for face detection and training general object detectors.

C++GeneralVisionData & viz

14.5k updated 3d ago

Deeplearning4jdeeplearning4j/deeplearning4j

Scalable deep learning for industry with parallel GPUs.

Java

14.3k updated 19d ago

SentencePiecegoogle/sentencepiece

A C++ library for unsupervised text tokenization and detokenization, widely used in modern NLP models.

C++NLP

12.2k updated 4d ago

segmentation_models.pytorchqubvel/segmentation_models.pytorch

A PyTorch-based toolkit that offers pre-trained segmentation models for computer vision tasks. It simplifies the development of image segmentation applications by providing a collection of popular architecture implementations, such as UNet and PSPNet, along with pre-trained weights, making it easier for researchers and developers to achieve high-quality pixel-level object segmentation in images.

PythonVision

11.8k updated 1d ago

Weights & Biaseswandb/wandb

Machine learning experiment tracking, dataset versioning, hyperparameter search, visualization, and collaboration

RLData & vizMLOps

11.3k updated today

Convnet.jskarpathy/convnetjs Deprecated

ConvNetJS is a JavaScript library for training Deep Learning models

JavaScriptGeneral

11.2k updated 3y ago

huggingface/tokenizershuggingface/tokenizers

Fast State-of-the-Art Tokenizers optimized for Research and Production

RustNLP

11.2k updated 1d ago

Turi Createapple/turicreate Archived

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.

PythonGeneralVision

11.2k archived

AutoGluonawslabs/autogluon

AutoML for Image, Text, Tabular, Time-Series, and MultiModal Data.

PythonGeneralVisionNLPMLOps

10.8k updated 2d ago

PyODyzhao062/pyod

> 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.

PythonGeneralLLMs

10.1k updated 12d ago

PySyftOpenMined/PySyft

A Python library for secure and private Deep Learning built on PyTorch and TensorFlow.

PythonGeneral

10k updated today

EspNetespnet/espnet

ESPnet is an end-to-end speech processing toolkit for tasks like speech recognition, translation, and enhancement, using PyTorch and Kaldi-style data processing.

PythonSpeech

10k updated today

einopsarogozhnikov/einops

Deep learning operations reinvented (for pytorch, tensorflow, jax and others).

PythonGeneral

9.6k updated 1mo ago

TFLearntflearn/tflearn

Deep learning library featuring a higher-level API for TensorFlow.

PythonGeneral

9.6k updated 2y ago

Hubactiveloopai/Hub

Fastest unstructured dataset management for TensorFlow/PyTorch. Stream & version-control data. Store even petabyte-scale data in a single numpy-like array on the cloud accessible on any machine. Visit activeloop.ai for more info.

PythonGeneralLLMsVisionMLOps

9.3k updated 5mo ago

BentoMLbentoml/bentoml

Toolkit for package and deploy machine learning models for serving in production

PythonGeneralLLMsMLOps

8.9k updated 1d ago