Pytorch Implementation Python Packages

pyiqa

🔎 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including PSNR, SSIM, LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, TOPIQ, NIMA, DBCNN, BRISQUE, PI and more...

metal-linalg

QR, symmetric eigendecomposition and SVD for batches of matrices on Apple Silicon GPUs, for MLX and PyTorch. 2x to 10x gain in performance. Metal kernels per regime, routed to the GPU or the CPU by policies measured per Mac. C++, Python (pip: metal-linalg, metal-linalg-torch), Swift and C.

zetascale

Build high-performance AI models with modular building blocks

sixdrepnet

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.

chuchichaestli

Where you find all the state-of-the-art cooking utensils (salt, pepper, gradient descent... the usual).

tversky

Differentiable Tversky similarity layers for PyTorch. Official implementation from the ICLR 2026 paper "Tversky Neural Networks."

metal-linalg-torch

QR, symmetric eigendecomposition and SVD for batches of matrices on Apple Silicon GPUs, for MLX and PyTorch. 2x to 10x gain in performance. Metal kernels per regime, routed to the GPU or the CPU by policies measured per Mac. C++, Python (pip: metal-linalg, metal-linalg-torch), Swift and C.

cerium-delta

Real-time neural network observability platform for monitoring activations, gradients, information flow, and architecture health.

dropblock

Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.

adfwi-torch

An Automatic Differentiation-based Waveform Inversion Framework Implemented in PyTorch.

dgmr

Implementation of DeepMind's Deep Generative Model of Radar (DGMR) https://arxiv.org/abs/2104.00954

fewsoldataloader

This repo hosts the PyTorch dataloader for FewSOL dataset | ICRA 2023

deep-fake-detection

This project Deepfakes can distort our perception of the truth and we need to develop a strategy to improve their detection. Deep Fakes are increasingly detrimental to privacy, social security, and democracy. We plan to achieve better accuracy in predicting real and fake videos.. https://pypi.org/project/Deep-Fake-Detection/

blitz-bayesian-pytorch

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.

alicia

A CLI to download, create, modify, train, test, predict and compare an image classifiers.

buctoolkit

Batch-upscaled Catalysis Toolkit

dynenv

Dynamic Simulation Environments for Reinforcement Learning

hift

memory-efficient fine-tuning; support 24G GPU memory fine-tuning 7B

causaltorch

CausalTorch is a PyTorch library for building generative models with explicit causal constraints. It integrates graph-based causal reasoning with deep learning to create AI systems that respect logical causal relationships.

onn

Online Deep Learning: Learning Deep Neural Networks on the Fly / Non-linear Contextual Bandit Algorithm (ONN_THS)

skillful-nowcasting

Implementation of DeepMind's Deep Generative Model of Radar (DGMR) https://arxiv.org/abs/2104.00954

aptx-neuron

This repository offers a Python code for the PyTorch implementation of the APTx Neuron, as introduced in the paper "APTx Neuron: A Unified Trainable Neuron Architecture Integrating Activation and Computation".

waveletnn

Orthonormal and Biorthogonal wavelet transform in PyTorch

kazane

Simple sinc interpolation in PyTorch.