pytorch-implementation

Translate manga/image 一键翻译各类图片内文字 https://cotrans.touhou.ai/ (no longer working)

A resource for learning about Machine learning & Deep Learning

A Collection of Variational Autoencoders (VAE) in PyTorch.

Tutorials on implementing a few sequence-to-sequence (seq2seq) models with PyTorch and TorchText.

🧠💬 Articles I wrote about machine learning, archived from MachineCurve.com.

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

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

This repository is the official implementation of Disentangling Writer and Character Styles for Handwriting Generation (CVPR 2023)

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

State-of-the-art, simple, fast unbounded / large-scale NeRFs.

[ACL 2024] Official PyTorch code for extracting features and training downstream models with emotion2vec: Self-Supervised Pre-Training for Speech Emotion Representation

LibCity: An Open Library for Urban Spatial-temporal Data Mining

Stable Diffusion implemented from scratch in PyTorch

A machine learning library with a TypeScript API and Rust backend. CUDA and WebGPU compatibility. Built to understand how ML frameworks and models work internally.

From scratch implementation of a sparse mixture of experts language model inspired by Andrej Karpathy's makemore :)

PyTorch Implementation of Physics-informed Neural Networks

Official implementation for "Generating Diverse and Natural 3D Human Motions from Texts (CVPR2022)."

Benchmark of federated learning. Dedicated to the community. 🤗

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

nnDetection is a self-configuring framework for 3D (volumetric) medical object detection which can be applied to new data sets without manual intervention. It includes guides for 12 data sets that were used to develop and evaluate the performance of the proposed method.

Build high-performance AI models with modular building blocks

Official Code for ECCV 2024 paper — One-Shot Diffusion Mimicker for Handwritten Text Generation

🦖Pytorch implementation of popular Attention Mechanisms, Vision Transformers, MLP-Like models and CNNs.🔥🔥🔥

The official PyTorch implementation of L2CS-Net for gaze estimation and tracking

KoBERT와 CRF로 만든 한국어 개체명인식기 (BERT+CRF based Named Entity Recognition model for Korean)