deep learning

Discover and compare deep learning tools for Python

PyTorch

80.0k

An open source machine learning framework that accelerates research to production

Machine Learning
Deep Learning
AI

TensorFlow

180.0k

An end-to-end open source machine learning platform

Machine Learning
Deep Learning
AI

Keras

60.0k

A deep learning API written in Python, running on top of TensorFlow

Machine Learning
Deep Learning
AI

MXNet

20.0k

A deep learning framework designed for both efficiency and flexibility

Machine Learning
Deep Learning
AI

JAX

22.0k

High-performance machine learning research tool for high-speed numerical computing

Machine Learning
Deep Learning
Numerical Computing

Theano

8.0k

A Python library to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays

Machine Learning
Deep Learning
Numerical Computing

Chainer

9.0k

A flexible framework for neural networks

Machine Learning
Deep Learning
AI

Transformers

120.0k

State-of-the-art NLP for PyTorch and TensorFlow 2.0

Natural Language Processing
Deep Learning
AI

AllenNLP

8.0k

An open-source NLP research library, built on PyTorch

Natural Language Processing
Deep Learning
Research