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

lpips

LPIPS metric. pip install lpips

dm-haiku

JAX-based neural network library

vosk

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

tf-nightly

An Open Source Machine Learning Framework for Everyone

neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

scikeras

Scikit-Learn API wrapper for Keras.

tensorflow-intel

An Open Source Machine Learning Framework for Everyone

onnx-weekly

Open standard for machine learning interoperability

tensorflow-cpu-aws

An Open Source Machine Learning Framework for Everyone

tensorflow-aarch64

An Open Source Machine Learning Framework for Everyone

pytabkit

ML models + benchmark for tabular data classification and regression

tensorflow-macos

An Open Source Machine Learning Framework for Everyone

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.

tf-nightly-cpu

An Open Source Machine Learning Framework for Everyone

tensorflow-gpu

An Open Source Machine Learning Framework for Everyone

ptflops

Flops counter for neural networks in pytorch framework

onnx-tf

Tensorflow Backend for ONNX

tensorly-torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch