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.

ncrnabert

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

neuralgcm-torch

PyTorch port of NeuralGCM

blitz-bayesian-pytorch

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

dinosaur-torch

PyTorch port of NeuralGCM

ghonn-models-pytorch

HONU based polynomial neural networks implemented in Pytorch

adfwi-torch

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

tversky

Official Pytorch Library for Tversky Neural Networks

dgmr

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

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/

model-suite

Implementations of various deep learning models in PyTorch

chuchichaestli

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

alicia

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

fewsoldataloader

Pytorch Dataloader for FewSOL

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.

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.

torch-metrics

Metrics for model evaluation in pytorch

ast-torch

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

explainable-cnn

📦 PyTorch based visualization package for generating layer-wise explanations for CNNs.

buctoolkit

BUCToolkit is a PyTorch-based high-performance AI4Science software package of computational chemistry,

potatorch

Lightweight high-level PyTorch framework that runs on potato machines