Neural Network Python Packages

torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

spacy

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

tensorflow

An Open Source Machine Learning Framework for Everyone

blis

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

onnx

Open standard for machine learning interoperability

gensim

Topic Modelling for Humans

ultralytics-thop

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

autograd

Efficiently computes derivatives of NumPy code.

kornia

🐍 Geometric Computer Vision Library for Spatial AI

paddlepaddle

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

tensorflow-cpu

An Open Source Machine Learning Framework for Everyone

torchtnt

A lightweight library for PyTorch training tools and utilities

mosaicml-streaming

A Data Streaming Library for Efficient Neural Network Training

tf-nightly

An Open Source Machine Learning Framework for Everyone

multi-model-server

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

model-archiver

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

onnx-weekly

Open standard for machine learning interoperability

pytorch-tokenizers

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

neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

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.

pytorch-ignite

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

executorch

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

nnaudio

Audio processing by using pytorch 1D convolution network

ncnn

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