pytorch-lightning

There are 26 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 8562
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 7526
Fanghua-Yu / SUPIR
SUPIR aims at developing Practical Algorithms for Photo-Realistic Image Restoration In the Wild. Our new online demo is also released at suppixel.ai.
deep-learning diffusion-models llava pytorch pytorch-lightning restoration sdxl stable-diffusion super-resolution
Language:Python 3990
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 3899
jdb78 / pytorch-forecasting
Time series forecasting with PyTorch
ai artifical-intelligense data-science deep-learning forecasting gpu machine-learning neural-networks pandas python pytorch pytorch-lightning temporal timeseries timeseries-forecasting uncertainty
Language:Python 3765
nerf_pl
kwea123 / nerf_pl
NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning
nerf neural-radiance-fields view-synthesis pytorch pytorch-lightning mesh 3d-reconstruction volume-rendering mixed-reality unity3d colab ray-tracing nerf-in-the-wild nerf-w
Language:Jupyter Notebook 2673
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 tutorials uvadlc pytorch pytorch-lightning tutorial flax jax optax
Language:Jupyter Notebook 2347
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 1952
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
classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric
Language:Python 1735
erikwijmans / Pointnet2_PyTorch
PyTorch implementation of Pointnet2/Pointnet++
hydra point-cloud pointnet2 pytorch pytorch-lightning
Language:Python 1454
vturrisi / solo-learn
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
simclr nvidia-dali contrastive-learning pytorch pytorch-lightning barlow-twins self-supervised-learning swav byol moco simsiam vicreg nnclr dino deepcluster ressl vibcreg transformer-models mae masked-input-prediction
Language:Python 1389
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 1353
manujosephv / pytorch_tabular
A standard framework for modelling Deep Learning Models for tabular data
tabular-data deep-learning pytorch pytorch-lightning hacktoberfest machine-learning
Language:Python 1273
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 1269
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 1235
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 894
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 842
OML-Team / open-metric-learning
Metric learning and retrieval pipelines, models and zoo.
computer-vision data-science deep-learning hacktoberfest hacktoberfest-2023 hacktoberfest2023 metric-learning pytorch pytorch-lightning representation-learning similarity-learning
Language:Python 840
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 671
norse
norse / norse
Deep learning with spiking neural networks (SNNs) in PyTorch.
spiking-neural-networks deep-learning pytorch tensor machine-learning gpu autograd neural-network neuromorphic pytorch-lightning
Language:Python 632
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 631
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 625
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.
pytorch-lightning remote-sensing-image vision-transformer deep-learning pytorch timm cnn segmentation semantic-segmentation
Language:Python 611
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)
face-swapping pytorch pytorch-lightning
Language:Python 606
torchmd / torchmd
End-To-End Molecular Dynamics (MD) Engine using PyTorch
molecular-dynamics coarse-grained-molecular-dynamics machine-learning neural-networks pytorch pytorch-lightning
Language:Python 534
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 424
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 422
krasserm / perceiver-io
A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training
deep-learning machine-learning perceiver perceiver-ar perceiver-io pytorch pytorch-lightning
Language:Python 419
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 407
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 395
ai4co / rl4co
A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)
attention attention-model benchmark combinatorial-optimization cvrp electronic-design-automation hydra neural-combinatorial-optimization operations-research pytorch-lightning reinforcement-learning scheduling tensordict torchrl tsp vehicle-routing-problem
Language:Python 341
astra-vision / SceneRF
[ICCV 2023] Official implementation of "SceneRF: Self-Supervised Monocular 3D Scene Reconstruction with Radiance Fields"
3d-reconstruction deep-learning depth-estimation nerf neural-radiance-fields pytorch pytorch-lightning scene-reconstruction self-supervised self-supervised-learning single-image-reconstruction
Language:Python 340
zjukg / NeuralKG
[Tool] For Knowledge Graph Representation Learning
python pytorch-lightning deep-learning machine-learning knowledge-graph knowledge-graph-embeddings link-prediction knowledge-graph-completion neuralkg
Language:Python 338
maum-ai / hififace
Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)
face-swapping pytorch pytorch-lightning wandb
Language:Python 337
ucuapps / OpenGlue
Open Source Graph Neural Net Based Pipeline for Image Matching
graph-neural-networks image-matching local-features kornia pytorch pytorch-lightning
Language:Python 331