Deep Neural Networks Python Packages

ctranslate2

Fast inference engine for Transformer models

onnx

Open standard for machine learning interoperability

tensorflow

An Open Source Machine Learning Framework for Everyone

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

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

scikeras

Scikit-Learn API wrapper for Keras.

dm-haiku

JAX-based neural network library

neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

tensorflow-cpu-aws

An Open Source Machine Learning Framework for Everyone

tensorflow-intel

An Open Source Machine Learning Framework for Everyone

pytabkit

ML models + benchmark for tabular data classification and regression

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

tensorflow-aarch64

An Open Source Machine Learning Framework for Everyone

ptflops

Flops counter for neural networks in pytorch framework

pytorch-tabnet

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

augraphy

Augmentation pipeline for rendering synthetic paper printing, faxing, scanning and copy machine processes

tensorly-torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch

tf-nightly-cpu

An Open Source Machine Learning Framework for Everyone

aimet-torch

AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.