pytorch-lightning

There are 28 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 9119
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 7749
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 sdxl stable-diffusion super-resolution restoration pytorch pytorch-lightning
Language:Python 5230
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 4880
sktime / 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 4504
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 2973
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 2800
labml
labmlai / labml
🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱
machine-learning deep-learning pytorch experiment analytics visualization tensorboard mobile keras tensorflow tensorflow2 keras-tensorflow pytorch-lightning fastai
Language:Python 2227
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 1740
erikwijmans / Pointnet2_PyTorch
PyTorch implementation of Pointnet2/Pointnet++
pointnet2 pytorch point-cloud pytorch-lightning hydra
Language:Python 1687
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 1568
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 1508
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 1433
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 1371
kwea123 / ngp_pl
Instant-ngp in pytorch+cuda trained with pytorch-lightning (high quality with high speed, with only few lines of legible code)
3d-reconstruction cuda instant-ngp nerf novel-view-synthesis pytorch pytorch-lightning
Language:Jupyter Notebook 1278
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 1104
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 951
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 919
airctic / icevision
An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come
ai annotation-parsers coco-dataset coco-parser computer-vision deep-learning effecientdet fastai faster-rcnn mask-rcnn object-detection pycocotools python pytorch pytorch-lightning tutorials voc-dataset voc-parser
Language:Python 866
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
3d-face-alignment 3d-face-reconstruction 3d-graphics 3d-reconstruction 3d-vision computer-vision cvpr deep-learning emotion-recognition face face-alignment face-model face-reconstruction flame-fitting morphable-model pytorch pytorch-lightning
Language:Python 787
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 719
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 657
grok-ai / nn-template
Generic template to bootstrap your PyTorch project.
best-practices best-practises cookiecutter deep-learning dvc github-actions huggingface-datasets hydra mkdocs pre-commit project-structure pytorch pytorch-lightning reproducibility research streamlit template wandb weights-and-biases
Language:Python 646
maum-ai / faceshifter
Unofficial PyTorch Implementation for FaceShifter (https://arxiv.org/abs/1912.13457)
pytorch pytorch-lightning face-swapping
Language:Python 631
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 611
Lightning-Universe / lightning-transformers
Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning
hydra pytorch pytorch-lightning transformers
Language:Python 609
ai4co / rl4co
A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)
combinatorial-optimization reinforcement-learning benchmark hydra tensordict torchrl pytorch-lightning operations-research electronic-design-automation attention-model cvrp tsp attention neural-combinatorial-optimization scheduling vehicle-routing-problem
Language:Python 550
JeffersonQin / YuzuMarker.FontDetection
✨ 首个CJK（中日韩）字体识别以及样式提取模型 YuzuMarker的字体识别模型与实现 / First-ever CJK (Chinese Japanese Korean) Font Recognition and Style Extractor, side project of YuzuMarker
cnn computer-vision cv font font-recognition fonts pytorch pytorch-cnn pytorch-lightning recognition chinese cjk-characters cjk-font japanese korean
Language:Python 477
mathLab / PINA
Physics-Informed Neural networks for Advanced modeling
physics-informed-neural-networks modeling machine-learning deep-learning python pytorch differential-equations ode pde physics-informed hacktoberfest lightining pinn equation-learning neural-networks neural-operators pytorch-lightning torch-geometric
Language:Python 468
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 461
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 457
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 430
machin
iffiX / machin
Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...
a3c-pytorch ddpg deep-learning distributed dqn ppo prioritized-experience-replay python pytorch pytorch-lightning pytorch-reinforcement-learning reinforcement-learning sac td3
Language:Python 407
mit-ll-responsible-ai / hydra-zen
Create powerful Hydra applications without the yaml files and boilerplate code.
configuration machine-learning dataclasses dynamic-configuration yaml-configuration pytorch pytorch-lightning reproducibility reproducible-science scalable configurable machinelearning deep-learning
Language:Python 370
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 368
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 365