PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Neural Network Python Packages

Python packages with the GitHub topic neural-network. Sorted by relevance, with stars and monthly downloads.
pytorch
torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

84.8M 100K 28K
tensorflow
tensorflow

An Open Source Machine Learning Framework for Everyone

22.6M 195K 75K
explosion
spacy

💫 Industrial-strength Natural Language Processing (NLP) in Python

22.1M 34K 5K
explosion
blis

💥 Fast matrix-multiplication as a self-contained Python library – no system dependencies!

20.9M 237 44
onnx
onnx

Open standard for machine learning interoperability

19.4M 21K 4K
ultralytics
ultralytics-thop

Profile PyTorch models for FLOPs and parameters, helping to evaluate computational efficiency and memory usage.

5.9M 140 8
RaRe-Technologies
gensim

Topic Modelling for Humans

5.1M 16K 4K
kornia
kornia

🐍 Geometric Computer Vision Library for Spatial AI

3.3M 11K 1K
paddlepaddle
paddlepaddle

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice (『飞桨』核心框架,深度学习&机器学习高性能单机、分布式训练和跨平台部署)

1.7M 24K 6K
tensorflow
tensorflow-cpu

An Open Source Machine Learning Framework for Everyone

1.4M 195K 75K
pytorch
torchtnt

A lightweight library for PyTorch training tools and utilities

697K 2K 300
mosaicml
mosaicml-streaming

A Data Streaming Library for Efficient Neural Network Training

517K 2K 188
tensorflow
tf-nightly

An Open Source Machine Learning Framework for Everyone

380K 195K 75K
PennyLaneAI
pennylane

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

336K 3K 778
Nixtla
neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

332K 4K 487
pytorch
pytorch-ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

310K 5K 696
tensorflow
tensorflow-intel

An Open Source Machine Learning Framework for Everyone

307K 195K 75K
onnx
onnx-weekly

Open standard for machine learning interoperability

272K 21K 4K
awslabs
multi-model-server

Multi Model Server is a tool for serving neural net models for inference

258K 1K 229
awslabs
model-archiver

Multi Model Server is a tool for serving neural net models for inference

247K 1K 229
pytorch
pytorch-tokenizers

On-device AI across mobile, embedded and edge for PyTorch

241K 5K 971
KinWaiCheuk
nnaudio

Audio processing by using pytorch 1D convolution network

239K 1K 97
Tencent
pnnx

ncnn is a high-performance neural network inference framework optimized for the mobile platform

238K 23K 4K
Tencent
ncnn

ncnn is a high-performance neural network inference framework optimized for the mobile platform

226K 23K 4K
    • Data from PyPI, GitHub, ClickHouse, and BigQuery