pytorch-lightning

There are 25 repositories under pytorch-lightning topic.

voicepaw / so-vits-svc-fork
so-vits-svc fork with realtime support, improved interface and more features.
sovits vits voice-conversion so-vits-svc hubert softvc realtime voice-changer deep-learning pytorch speech-synthesis contentvec gan lightning pytorch-lightning hacktoberfest
Language:Python 8375
XavierXiao / Dreambooth-Stable-Diffusion
Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion
pytorch pytorch-lightning stable-diffusion text-to-image
Language:Jupyter Notebook 7464
ashleve / lightning-hydra-template
PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡
pytorch-lightning project-structure config deep-learning hydra template pytorch reproducibility best-practices mlops
Language:Python 3709
jdb78 / pytorch-forecasting
Time series forecasting with PyTorch
pytorch forecasting gpu uncertainty timeseries-forecasting pytorch-lightning deep-learning neural-networks timeseries machine-learning pandas python ai artifical-intelligense data-science temporal
Language:Python 3641
Fanghua-Yu / SUPIR
SUPIR aims at developing Practical Algorithms for Photo-Realistic Image Restoration In the Wild
deep-learning diffusion-models llava sdxl stable-diffusion super-resolution restoration pytorch pytorch-lightning
Language:Python 3513
nerf_pl
kwea123 / nerf_pl
NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning
3d-reconstruction colab mesh mixed-reality nerf nerf-in-the-wild nerf-w neural-radiance-fields pytorch pytorch-lightning ray-tracing unity3d view-synthesis volume-rendering
Language:Jupyter Notebook 2635
phlippe / uvadlc_notebooks
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
deep-learning flax jax optax pytorch pytorch-lightning tutorial tutorials uvadlc
Language:Jupyter Notebook 2175
labml
labmlai / labml
🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱
analytics deep-learning experiment fastai keras keras-tensorflow machine-learning mobile pytorch pytorch-lightning tensorboard tensorflow tensorflow2 visualization
Language:Jupyter Notebook 1875
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
pytorch-lightning deep-learning machine-learning pytorch tasks-flash classification tabular-data object-detection icevision open3d fiftyone torch-geometric pytorch-video
Language:Python 1729
erikwijmans / Pointnet2_PyTorch
PyTorch implementation of Pointnet2/Pointnet++
hydra point-cloud pointnet2 pytorch pytorch-lightning
Language:Python 1403
vturrisi / solo-learn
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
barlow-twins byol contrastive-learning deepcluster dino mae masked-input-prediction moco nnclr nvidia-dali pytorch pytorch-lightning ressl self-supervised-learning simclr simsiam swav transformer-models vibcreg vicreg
Language:Python 1359
traingenerator
jrieke / traingenerator
🧙 A web app to generate template code for machine learning
machine-learning deep-learning template website webapp streamlit pytorch scikit-learn sklearn tensorflow pytorch-ignite machine-learning-tutorials pytorch-lightning
Language:Python 1345
manujosephv / pytorch_tabular
A standard framework for modelling Deep Learning Models for tabular data
deep-learning hacktoberfest machine-learning pytorch pytorch-lightning tabular-data
Language:Python 1225
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.
best-practices config deep-learning pytorch pytorch-lightning pytorch-lightning-template template
Language:Jupyter Notebook 1217
kwea123 / ngp_pl
Instant-ngp in pytorch+cuda trained with pytorch-lightning (high quality with high speed, with only few lines of legible code)
instant-ngp cuda nerf pytorch-lightning 3d-reconstruction pytorch novel-view-synthesis
Language:Jupyter Notebook 1216
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.
bert bert-model huggingface-transformers huggingface nlp toxic-comment-classification toxicity toxic-comments sentence-classification kaggle-competition pytorch-lightning hatespeech hate-speech-detection toxicity-classification hate-speech
Language:Python 846
airctic / icevision
An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come
object-detection deep-learning pytorch pytorch-lightning python fastai ai computer-vision effecientdet faster-rcnn mask-rcnn annotation-parsers tutorials coco-parser voc-parser pycocotools coco-dataset voc-dataset
Language:Python 840
OML-Team / open-metric-learning
Library for metric learning pipelines and models.
computer-vision data-science deep-learning hacktoberfest hacktoberfest-2023 hacktoberfest2023 metric-learning pytorch pytorch-lightning representation-learning similarity-learning
Language:Jupyter Notebook 775
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
computer-vision face-reconstruction emotion-recognition cvpr pytorch pytorch-lightning deep-learning 3d-reconstruction 3d-graphics 3d-vision face-alignment morphable-model face-model flame-fitting face 3d-face-alignment 3d-face-reconstruction
Language:Python 648
quaterion
qdrant / quaterion
Blazing fast framework for fine-tuning similarity learning models
contrastive-learning cosine-similarity deep-learning knn machine-learning metric-learning nearest-neighbor-search python pytorch pytorch-lightning similarity-learning similarity-search
Language:Python 625
grok-ai / nn-template
Generic template to bootstrap your PyTorch project.
pytorch template pytorch-lightning wandb dvc hydra research deep-learning best-practices reproducibility project-structure weights-and-biases streamlit mkdocs github-actions best-practises pre-commit cookiecutter huggingface-datasets
Language:Python 619
norse
norse / norse
Deep learning with spiking neural networks (SNNs) in PyTorch.
autograd deep-learning gpu machine-learning neural-network neuromorphic pytorch pytorch-lightning spiking-neural-networks tensor
Language:Python 616
Lightning-Universe / lightning-transformers
Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning
pytorch pytorch-lightning transformers hydra
Language:Python 607
maum-ai / faceshifter
Unofficial PyTorch Implementation for FaceShifter (https://arxiv.org/abs/1912.13457)
pytorch pytorch-lightning face-swapping
Language:Python 592
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.
cnn deep-learning pytorch pytorch-lightning remote-sensing-image segmentation semantic-segmentation timm vision-transformer
Language:Python 548
torchmd / torchmd
End-To-End Molecular Dynamics (MD) Engine using PyTorch
coarse-grained-molecular-dynamics machine-learning molecular-dynamics neural-networks pytorch pytorch-lightning
Language:Python 511
TransformerSum
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.
automatic-summarization machine-learning extractive-summarization bert roberta distilbert albert summarization transformer-models pytorch-lightning summarization-dataset text-summarization
Language:Python 425
krasserm / perceiver-io
A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training
perceiver deep-learning machine-learning pytorch pytorch-lightning perceiver-io perceiver-ar
Language:Python 408
JeffersonQin / YuzuMarker.FontDetection
✨ 首个CJK（中日韩）字体识别以及样式提取模型 YuzuMarker的字体识别模型与实现 / First-ever CJK (Chinese Japanese Korean) Font Recognition and Style Extractor, side project of YuzuMarker
chinese cjk-characters cjk-font cnn computer-vision cv font font-recognition fonts japanese korean pytorch pytorch-cnn pytorch-lightning recognition
Language:Python 406
cccntu / minLoRA
minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.
fastai huggingface pytorch pytorch-implementation pytorch-lightning
Language:Jupyter Notebook 392
machin
iffiX / machin
Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...
reinforcement-learning deep-learning pytorch pytorch-reinforcement-learning dqn ddpg sac ppo td3 prioritized-experience-replay distributed a3c-pytorch python pytorch-lightning
Language:Python 389
astra-vision / SceneRF
[ICCV 2023] Official implementation of "SceneRF: Self-Supervised Monocular 3D Scene Reconstruction with Radiance Fields"
deep-learning nerf neural-radiance-fields pytorch pytorch-lightning scene-reconstruction single-image-reconstruction depth-estimation self-supervised-learning 3d-reconstruction self-supervised
Language:Python 332
maum-ai / hififace
Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)
pytorch face-swapping pytorch-lightning wandb
Language:Python 331
ecs-vlc / FMix
Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'
augmentation pytorch pytorch-lightning torchbearer tensorflow
Language:Jupyter Notebook 330
ucuapps / OpenGlue
Open Source Graph Neural Net Based Pipeline for Image Matching
graph-neural-networks image-matching kornia local-features pytorch pytorch-lightning
Language:Python 326
VITA-Group / SinNeRF
[ECCV 2022] "SinNeRF: Training Neural Radiance Fields on Complex Scenes from a Single Image", Dejia Xu, Yifan Jiang, Peihao Wang, Zhiwen Fan, Humphrey Shi, Zhangyang Wang
nerf pytorch pytorch-lightning gan neural-radiance-fields
Language:Python 324