#Pytorch-lightning
Showing 55 of 55 repositories tagged #pytorch-lightning, ranked by stars
so-vits-svc fork with realtime support, improved interface and more features.
SUPIR aims at developing Practical Algorithms for Photo-Realistic Image Restoration In the Wild. Our new online demo is also released at suppixel.ai.
PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡
Time series forecasting with PyTorch
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
A unified framework for Deep Learning Models on tabular data
An easy/swift-to-adapt PyTorch-Lighting template. 套壳模板，简单易用，稍改原来Pytorch代码，即可适配Lightning。You can translate your previous Pytorch code much easier using this template, and keep your freedom to edit all the functions as well. Big-project-friendly as well. No need to rewrite your config in hydra.
🧙 A web app to generate template code for machine learning
Metric learning and retrieval pipelines, models and zoo.
An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come
A minimalist SOTA LaTeX OCR model with only 20M parameters, running in browser. Full training pipeline available for self-reproduction. | 超轻量SOTA LaTeX公式识别模型，仅20M参数量，可在浏览器中运行。训练全流程代码开源，以便自学复现。
Official repository accompanying a CVPR 2022 paper EMOCA: Emotion Driven Monocular Face Capture And Animation. EMOCA takes a single image of a face as input and produces a 3D reconstruction. EMOCA sets the new standard on reconstructing highly emotional images in-the-wild
Deep learning with spiking neural networks (SNNs) in PyTorch.
Physics-Informed Neural networks for Advanced modeling
End-To-End Molecular Dynamics (MD) Engine using PyTorch
Blazing fast framework for fine-tuning similarity learning models
Generic template to bootstrap your PyTorch project.
✨ 首个CJK（中日韩）字体识别以及样式提取模型 YuzuMarker的字体识别模型与实现 / First-ever CJK (Chinese Japanese Korean) Font Recognition and Style Extractor, side project of YuzuMarker
A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training
Create powerful Hydra applications without the yaml files and boilerplate code.
Library for automatic retraining and continual learning
Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).
Examples of python neural net and ML stock prediction methods with sample stock data.
A deep learning model to age faces in the wild, currently runs at 60+ fps on GPUs
State-of-the-art model for person re-identification in Multi-camera Multi-Target Tracking. Benchmarked on Market-1501 and DukeMTMTC-reID datasets.
Annotated Flow Matching paper
A lightweight runtime health check for PyTorch training runs.
Neural HMMs are all you need (for high-quality attention-free TTS)
A Python package for identifying hundreds of kinds of animals, training custom models, and estimating distance from camera trap videos and images
Educational implementation of the Discrete Flow Matching paper
PyTorch code for the paper "You Don’t Need Strong Assumptions: Visual Representation Learning via Temporal Differences"
[ICCV 2023] Multi3DRefer: Grounding Text Description to Multiple 3D Objects
Pure PyTorch implementation of Nvidia's hash grid encoding: https://nvlabs.github.io/instant-ngp/
2D keypoint detection with Pytorch Lightning and wandb
👦 Human head semantic segmentation
classy is a simple-to-use library for building high-performance Machine Learning models in NLP.
Time series forecasting with PyTorch
Wheat Detection challenge on Kaggle
A baseline segmentation example using the comma10k dataset (WIP)
Code Accompany NeurIPS 2022 Dataset Paper: Breaking Bad. A Shape Assembly Benchmark Implemented in PyTorch
📡 PyTorch Lightning Implementations of Recent Satellite Image Classification !
The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends
⚡️ Implementation of TRON: Transformer Recommender using Optimized Negative-sampling, accepted at ACM RecSys 2023.
A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.
Automatic machine learning for tabular data. ⚡🔥⚡
Use QLoRA to tune LLM in PyTorch-Lightning w/ Huggingface + MLflow
Generative models (GAN, VAE, Diffusion Models, Autoregressive Models) implemented with Pytorch, Pytorch_lightning and hydra.
🤹 MultiTRON: Pareto Front Approximation for Multi-Objective Session-Based Recommender Systems, accepted at ACM RecSys 2024.
PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/implementations - Vision Transformer (ViT), ResNetV2, EfficientNetV2, NeRF, SegFormer, MixTransformer, (planned...) DeepLabV3+, ConvNeXtV2, YOLO, etc.
This is a repository of neural full-rank spatial covariance analysis with speaker activity (neural FCASA).
A docker-based starter kit for machine learning via jupyter notebooks. Designed for those who just want a runtime environment and get on with machine learning. Docker Hub:
Pytorch implementation of GEE: A Gradient-based Explainable Variational Autoencoder for Network Anomaly Detection
Official implementation of Generative Colorization of Structured Mobile Web Pages, WACV 2023.