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...

zetascale

Build high-performance AI models with modular building blocks

sixdrepnet

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

dropblock

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

chuchichaestli

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

deep-fake-detection

Deep Fake Detection Project

ncrnabert

ncRNABert: Deciphering the landscape of non-coding RNA using language model

blitz-bayesian-pytorch

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

adfwi-torch

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

dgmr

PyTorch Skillful Nowcasting GAN Implementation

fewsoldataloader

Pytorch Dataloader for FewSOL

modtorch

MODTORCH is a powerful meta-language for building PyTorch networks on the fly, without having to write custom PyTorch classes manually.

buctoolkit

Batch-upscaled Catalysis Toolkit

ghonn-models-pytorch

HONU based polynomial neural networks implemented in Pytorch

alicia

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

model-suite

Model Suite

torch-metrics

Metrics for model evaluation in pytorch

causaltorch

A PyTorch library for building generative models with causal constraints

ast-torch

Implementation of AST from the paper: "AST: Audio Spectrogram Transformer' in PyTorch and Zeta

timediffusion

TimeDiffusion - Unified Framework for Multiple Time Series Tasks

torchsom

TorchSOM is a PyTorch-based library for training Self-Organizing Maps (SOMs), a model trained in an unsupervised manner, that can be used for clustering, dimensionality reduction and data visualization. It is designed to be scalable and user-friendly.

explainable-cnn

📦 Flexible visualization package for generatinglayer-wise explanations for CNNs.

dynenv

Dynamic Simulation Environments for Reinforcement Learning

hift

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