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

tensorflow-cpu

An Open Source Machine Learning Framework for Everyone

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

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

onnx-weekly

Open standard for machine learning interoperability

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

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

pytorch-tabnet

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

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

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

tensorly-torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch