pytorch-lightning
There are 953 repositories under pytorch-lightning topic.
voicepaw/so-vits-svc-fork
so-vits-svc fork with realtime support, improved interface and more features.
XavierXiao/Dreambooth-Stable-Diffusion
Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion
ashleve/lightning-hydra-template
PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡
jdb78/pytorch-forecasting
Time series forecasting with PyTorch
Fanghua-Yu/SUPIR
SUPIR aims at developing Practical Algorithms for Photo-Realistic Image Restoration In the Wild
kwea123/nerf_pl
NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning
phlippe/uvadlc_notebooks
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
labmlai/labml
🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱
Lightning-Universe/lightning-flash
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
erikwijmans/Pointnet2_PyTorch
PyTorch implementation of Pointnet2/Pointnet++
vturrisi/solo-learn
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
jrieke/traingenerator
🧙 A web app to generate template code for machine learning
manujosephv/pytorch_tabular
A standard framework for modelling Deep Learning Models for tabular data
miracleyoo/pytorch-lightning-template
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.
kwea123/ngp_pl
Instant-ngp in pytorch+cuda trained with pytorch-lightning (high quality with high speed, with only few lines of legible code)
unitaryai/detoxify
Trained models & code to predict toxic comments on all 3 Jigsaw Toxic Comment Challenges. Built using ⚡ Pytorch Lightning and 🤗 Transformers. For access to our API, please email us at contact@unitary.ai.
airctic/icevision
An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come
OML-Team/open-metric-learning
Library for metric learning pipelines and models.
radekd91/emoca
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
qdrant/quaterion
Blazing fast framework for fine-tuning similarity learning models
norse/norse
Deep learning with spiking neural networks (SNNs) in PyTorch.
grok-ai/nn-template
Generic template to bootstrap your PyTorch project.
Lightning-Universe/lightning-transformers
Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning
maum-ai/faceshifter
Unofficial PyTorch Implementation for FaceShifter (https://arxiv.org/abs/1912.13457)
WangLibo1995/GeoSeg
UNetFormer: A UNet-like transformer for efficient semantic segmentation of remote sensing urban scene imagery, ISPRS. Also, including other vision transformers and CNNs for satellite, aerial image and UAV image segmentation.
torchmd/torchmd
End-To-End Molecular Dynamics (MD) Engine using PyTorch
HHousen/TransformerSum
Models to perform neural summarization (extractive and abstractive) using machine learning transformers and a tool to convert abstractive summarization datasets to the extractive task.
krasserm/perceiver-io
A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training
JeffersonQin/YuzuMarker.FontDetection
✨ 首个CJK（中日韩）字体识别以及样式提取模型 YuzuMarker的字体识别模型与实现 / First-ever CJK (Chinese Japanese Korean) Font Recognition and Style Extractor, side project of YuzuMarker
cccntu/minLoRA
minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.
iffiX/machin
Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...
maum-ai/hififace
Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)
astra-vision/SceneRF
[ICCV 2023] Official implementation of "SceneRF: Self-Supervised Monocular 3D Scene Reconstruction with Radiance Fields"
ecs-vlc/FMix
Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'
ucuapps/OpenGlue
Open Source Graph Neural Net Based Pipeline for Image Matching
zjukg/NeuralKG
[Tool] For Knowledge Graph Representation Learning