About PyTorch

An open source machine learning framework that accelerates research to production

uv add pytorch

TensorFlow

An end-to-end open source machine learning platform

194.0k stars

Keras

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

63.9k stars

MXNet

A deep learning framework designed for both efficiency and flexibility

20.8k stars

Chainer

A flexible framework for neural networks

5.9k stars

pytools.dev