Deep Neural Networks Python Packages

tensorflow

An Open Source Machine Learning Framework for Everyone

onnx

Open standard for machine learning interoperability

ctranslate2

Fast inference engine for Transformer models

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

tensorflow-cpu

An Open Source Machine Learning Framework for Everyone

dm-haiku

JAX-based neural network library

lpips

LPIPS metric. pip install lpips

tf-nightly

An Open Source Machine Learning Framework for Everyone

vosk

Offline speech recognition API for Android, iOS, Raspberry Pi and servers with Python, Java, C# and Node

neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

onnx-weekly

Open standard for machine learning interoperability

tensorflow-intel

An Open Source Machine Learning Framework for Everyone

scikeras

Scikit-Learn API wrapper for Keras.

tensorflow-cpu-aws

An Open Source Machine Learning Framework for Everyone

tensorflow-aarch64

An Open Source Machine Learning Framework for Everyone

tensorflow-macos

An Open Source Machine Learning Framework for Everyone

ptflops

Flops counter for neural networks in pytorch framework

tf-nightly-cpu

An Open Source Machine Learning Framework for Everyone

pytabkit

ML models + benchmark for tabular data classification and regression

pytorch-tabnet

PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf

torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

tensorflow-gpu

An Open Source Machine Learning Framework for Everyone

onnx-tf

Tensorflow Backend for ONNX

airborne-antara

Production-grade adaptive meta-learning framework for continual model improvement. Implements research DOI: 10.5281/zenodo.17839490.