pytorch-lightning | Topic | Ecosyste.ms: Repos

Topic: "pytorch-lightning"

voicepaw/so-vits-svc-fork

so-vits-svc fork with realtime support, improved interface and more features.

Language: Python - Size: 20.2 MB - Last synced at: 5 days ago - Pushed at: 22 days ago - Stars: 9,058 - Forks: 1,205

XavierXiao/Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion

Language: Jupyter Notebook - Size: 5.71 MB - Last synced at: about 2 months ago - Pushed at: over 2 years ago - Stars: 7,725 - Forks: 801

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.

Language: Python - Size: 10.2 MB - Last synced at: about 2 months ago - Pushed at: 2 months ago - Stars: 5,069 - Forks: 432

ashleve/lightning-hydra-template

PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡

Language: Python - Size: 3.6 MB - Last synced at: about 2 months ago - Pushed at: 11 months ago - Stars: 4,663 - Forks: 693

sktime/pytorch-forecasting

Time series forecasting with PyTorch

Language: Python - Size: 37.5 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 4,378 - Forks: 689

kwea123/nerf_pl

NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning

Language: Jupyter Notebook - Size: 22.8 MB - Last synced at: about 2 months ago - Pushed at: almost 2 years ago - Stars: 2,791 - Forks: 479

phlippe/uvadlc_notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

Language: Jupyter Notebook - Size: 158 MB - Last synced at: 4 months ago - Pushed at: 4 months ago - Stars: 2,752 - Forks: 615

labmlai/labml

🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱

Language: Python - Size: 101 MB - Last synced at: 24 days ago - Pushed at: 3 months ago - Stars: 2,173 - Forks: 143

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

Language: Python - Size: 12.8 MB - Last synced at: 6 days ago - Pushed at: almost 2 years ago - Stars: 1,741 - Forks: 211

erikwijmans/Pointnet2_PyTorch

PyTorch implementation of Pointnet2/Pointnet++

Language: Python - Size: 953 KB - Last synced at: about 2 months ago - Pushed at: 4 months ago - Stars: 1,637 - Forks: 361

manujosephv/pytorch_tabular

A standard framework for modelling Deep Learning Models for tabular data

Language: Python - Size: 39.6 MB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 1,546 - Forks: 153

vturrisi/solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

Language: Python - Size: 5.04 MB - Last synced at: 11 days ago - Pushed at: about 1 month ago - Stars: 1,493 - Forks: 192

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.

Language: Jupyter Notebook - Size: 249 KB - Last synced at: about 2 months ago - Pushed at: almost 2 years ago - Stars: 1,458 - Forks: 195

jrieke/traingenerator

🧙 A web app to generate template code for machine learning

Language: Python - Size: 9.86 MB - Last synced at: 1 day ago - Pushed at: almost 2 years ago - Stars: 1,373 - Forks: 186

kwea123/ngp_pl

Instant-ngp in pytorch+cuda trained with pytorch-lightning (high quality with high speed, with only few lines of legible code)

Language: Jupyter Notebook - Size: 14.7 MB - Last synced at: about 2 months ago - Pushed at: about 2 years ago - Stars: 1,279 - Forks: 157

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 [email protected].

Language: Python - Size: 50.8 MB - Last synced at: 17 days ago - Pushed at: 3 months ago - Stars: 1,071 - Forks: 127

OML-Team/open-metric-learning

Metric learning and retrieval pipelines, models and zoo.

Language: Python - Size: 2.51 MB - Last synced at: about 2 months ago - Pushed at: 3 months ago - Stars: 952 - Forks: 71

airctic/icevision

An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come

Language: Python - Size: 848 MB - Last synced at: 18 days ago - Pushed at: 8 months ago - Stars: 858 - Forks: 149

norse/norse

Deep learning with spiking neural networks (SNNs) in PyTorch.

Language: Python - Size: 110 MB - Last synced at: about 19 hours ago - Pushed at: about 23 hours ago - Stars: 742 - Forks: 91

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

Language: Python - Size: 4.57 MB - Last synced at: 7 months ago - Pushed at: 7 months ago - Stars: 725 - Forks: 90

qdrant/quaterion

Blazing fast framework for fine-tuning similarity learning models

Language: Python - Size: 5.15 MB - Last synced at: 23 days ago - Pushed at: 3 months ago - Stars: 655 - Forks: 45

torchmd/torchmd

End-To-End Molecular Dynamics (MD) Engine using PyTorch

Language: Python - Size: 2.51 MB - Last synced at: 18 days ago - Pushed at: 6 months ago - Stars: 645 - Forks: 80

grok-ai/nn-template

Generic template to bootstrap your PyTorch project.

Language: Python - Size: 2.68 MB - Last synced at: about 2 months ago - Pushed at: almost 2 years ago - Stars: 641 - Forks: 67

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.

Language: Python - Size: 1.3 MB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 640 - Forks: 104

maum-ai/faceshifter

Unofficial PyTorch Implementation for FaceShifter (https://arxiv.org/abs/1912.13457)

Language: Python - Size: 96.8 MB - Last synced at: about 2 months ago - Pushed at: over 3 years ago - Stars: 635 - Forks: 120

Lightning-Universe/lightning-transformers 📦

Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

Language: Python - Size: 825 KB - Last synced at: 11 days ago - Pushed at: over 2 years ago - Stars: 609 - Forks: 74

ai4co/rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

Language: Python - Size: 155 MB - Last synced at: 14 days ago - Pushed at: 14 days ago - Stars: 607 - Forks: 111

mathLab/PINA

Physics-Informed Neural networks for Advanced modeling

Language: Python - Size: 133 MB - Last synced at: 4 days ago - Pushed at: 7 days ago - Stars: 528 - Forks: 81

JeffersonQin/YuzuMarker.FontDetection

✨ 首个CJK（中日韩）字体识别以及样式提取模型 YuzuMarker的字体识别模型与实现 / First-ever CJK (Chinese Japanese Korean) Font Recognition and Style Extractor, side project of YuzuMarker

Language: Python - Size: 172 KB - Last synced at: about 2 months ago - Pushed at: 5 months ago - Stars: 485 - Forks: 23

krasserm/perceiver-io

A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training

Language: Python - Size: 22.2 MB - Last synced at: 21 days ago - Pushed at: over 1 year ago - Stars: 479 - Forks: 43

changjonathanc/minLoRA

minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.

Language: Jupyter Notebook - Size: 15.6 KB - Last synced at: about 1 month ago - Pushed at: about 2 years ago - Stars: 464 - Forks: 30

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.

Language: Python - Size: 11.7 MB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 431 - Forks: 57

iffiX/machin

Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...

Language: Python - Size: 6.66 MB - Last synced at: 17 days ago - Pushed at: almost 4 years ago - Stars: 413 - Forks: 51

hmorimitsu/ptlflow

PyTorch Lightning Optical Flow models, scripts, and pretrained weights.

Language: Python - Size: 9.56 MB - Last synced at: 13 days ago - Pushed at: 13 days ago - Stars: 388 - Forks: 46

mit-ll-responsible-ai/hydra-zen

Create powerful Hydra applications without the yaml files and boilerplate code.

Language: Python - Size: 23.5 MB - Last synced at: 11 days ago - Pushed at: 15 days ago - Stars: 388 - Forks: 17

zjukg/NeuralKG

[Tool] For Knowledge Graph Representation Learning

Language: Python - Size: 62.8 MB - Last synced at: 2 months ago - Pushed at: over 1 year ago - Stars: 371 - Forks: 66

maum-ai/hififace

Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)

Language: Python - Size: 14.3 MB - Last synced at: 19 days ago - Pushed at: over 2 years ago - Stars: 368 - Forks: 59

ecs-vlc/FMix

Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'

Language: Jupyter Notebook - Size: 1.12 MB - Last synced at: 2 months ago - Pushed at: over 4 years ago - Stars: 335 - Forks: 38

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

Language: Python - Size: 53.4 MB - Last synced at: about 2 months ago - Pushed at: almost 3 years ago - Stars: 331 - Forks: 26

astra-vision/SceneRF

[ICCV 2023] Official implementation of "SceneRF: Self-Supervised Monocular 3D Scene Reconstruction with Radiance Fields"

Language: Python - Size: 220 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 313 - Forks: 26

ucuapps/OpenGlue

Open Source Graph Neural Net Based Pipeline for Image Matching

Language: Python - Size: 324 KB - Last synced at: over 1 year ago - Pushed at: almost 2 years ago - Stars: 305 - Forks: 41

awslabs/Renate

Library for automatic retraining and continual learning

Language: Python - Size: 6.58 MB - Last synced at: 20 days ago - Pushed at: 10 months ago - Stars: 294 - Forks: 6

kwea123/CasMVSNet_pl

Cascade Cost Volume for High-Resolution Multi-View Stereo and Stereo Matching using pytorch-lightning

Language: Jupyter Notebook - Size: 39.5 MB - Last synced at: about 2 months ago - Pushed at: over 1 year ago - Stars: 290 - Forks: 31

D-dot-AT/Stock-Prediction-Neural-Network-and-Machine-Learning-Examples

Examples of python neural net and ML stock prediction methods with sample stock data.

Language: Python - Size: 15.3 MB - Last synced at: 8 months ago - Pushed at: 8 months ago - Stars: 269 - Forks: 59

paninski-lab/lightning-pose

Accelerated pose estimation and tracking using semi-supervised learning

Language: Python - Size: 141 MB - Last synced at: 4 days ago - Pushed at: 5 days ago - Stars: 267 - Forks: 44

mx-mark/VideoTransformer-pytorch

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.

Language: Python - Size: 4.17 MB - Last synced at: about 1 year ago - Pushed at: about 3 years ago - Stars: 257 - Forks: 35

HasnainRaz/Fast-AgingGAN

A deep learning model to age faces in the wild, currently runs at 60+ fps on GPUs

Language: Python - Size: 9.92 MB - Last synced at: about 2 months ago - Pushed at: about 1 year ago - Stars: 256 - Forks: 49

openclimatefix/skillful_nowcasting

Implementation of DeepMind's Deep Generative Model of Radar (DGMR) https://arxiv.org/abs/2104.00954

Language: Python - Size: 195 KB - Last synced at: about 17 hours ago - Pushed at: about 22 hours ago - Stars: 253 - Forks: 61

Unbabel/OpenKiwi

Open-Source Machine Translation Quality Estimation in PyTorch

Language: Python - Size: 33.6 MB - Last synced at: 20 days ago - Pushed at: about 3 years ago - Stars: 231 - Forks: 47

CompVis/net2net

Network-to-Network Translation with Conditional Invertible Neural Networks

Language: Python - Size: 75.2 MB - Last synced at: 20 days ago - Pushed at: over 2 years ago - Stars: 225 - Forks: 21

kwea123/nsff_pl

Neural Scene Flow Fields using pytorch-lightning, with potential improvements

Language: Jupyter Notebook - Size: 51.3 MB - Last synced at: 2 months ago - Pushed at: over 2 years ago - Stars: 225 - Forks: 28

gorodnitskiy/yet-another-lightning-hydra-template

Flexible and scalable template based on PyTorch Lightning + Hydra. Efficient workflow and reproducibility for rapid ML experiments.

Language: Python - Size: 229 KB - Last synced at: 3 months ago - Pushed at: over 1 year ago - Stars: 221 - Forks: 11

Erlemar/pytorch_tempest

My repo for training neural nets using pytorch-lightning and hydra

Language: Python - Size: 114 MB - Last synced at: 2 months ago - Pushed at: 5 months ago - Stars: 220 - Forks: 21

SurajDonthi/Multi-Camera-Person-Re-Identification

State-of-the-art model for person re-identification in Multi-camera Multi-Target Tracking. Benchmarked on Market-1501 and DukeMTMTC-reID datasets.

Language: Python - Size: 1.49 MB - Last synced at: 4 months ago - Pushed at: over 1 year ago - Stars: 217 - Forks: 36

karanchahal/distiller

A large scale study of Knowledge Distillation.

Language: Python - Size: 717 MB - Last synced at: about 1 year ago - Pushed at: about 5 years ago - Stars: 215 - Forks: 29

junhsss/consistency-models

A Toolkit for OpenAI's Consistency Models.

Language: Python - Size: 55.1 MB - Last synced at: about 2 months ago - Pushed at: over 2 years ago - Stars: 205 - Forks: 13

Rose-STL-Lab/dyffusion

[NeurIPS 2023] A Dynamics-informed Diffusion Model for Spatiotemporal Forecasting

Language: Python - Size: 521 KB - Last synced at: about 2 months ago - Pushed at: 9 months ago - Stars: 202 - Forks: 22

lightning-uq-box/lightning-uq-box

Lightning-UQ-Box: Uncertainty Quantification for Neural Networks with PyTorch and Lightning

Language: Python - Size: 47 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 186 - Forks: 21

florencejt/fusilli

A Python package housing a collection of deep-learning multi-modal data fusion method pipelines! From data loading, to training, to evaluation - fusilli's got you covered 🌸

Language: Python - Size: 987 MB - Last synced at: 13 days ago - Pushed at: 13 days ago - Stars: 186 - Forks: 14

lebellig/flow-matching

Annotated Flow Matching paper

Language: Jupyter Notebook - Size: 290 KB - Last synced at: 3 months ago - Pushed at: 10 months ago - Stars: 182 - Forks: 8

PinataFarms/FEARTracker

Official repo for FEAR: Fast, Efficient, Accurate and Robust Visual Tracker (ECCV 2022)

Language: Python - Size: 34.7 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 176 - Forks: 17

IntelLabs/matsciml

Open MatSci ML Toolkit is a framework for prototyping and scaling out deep learning models for materials discovery supporting widely used materials science datasets, and built on top of PyTorch Lightning, the Deep Graph Library, and PyTorch Geometric.

Language: Python - Size: 35.9 MB - Last synced at: 7 days ago - Pushed at: 4 months ago - Stars: 175 - Forks: 29

nocotan/pytorch-lightning-gans

Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.

Language: Python - Size: 42 KB - Last synced at: 6 months ago - Pushed at: over 3 years ago - Stars: 167 - Forks: 16

shivammehta25/Neural-HMM

Neural HMMs are all you need (for high-quality attention-free TTS)

Language: Jupyter Notebook - Size: 76.3 MB - Last synced at: 22 days ago - Pushed at: 22 days ago - Stars: 158 - Forks: 25

logancyang/loss-landscape-anim

Create animations for the optimization trajectory of neural nets

Language: Python - Size: 71.6 MB - Last synced at: about 2 months ago - Pushed at: over 1 year ago - Stars: 153 - Forks: 24

autonlab/weasel

Weakly Supervised End-to-End Learning (NeurIPS 2021)

Language: Python - Size: 26.5 MB - Last synced at: over 1 year ago - Pushed at: over 2 years ago - Stars: 153 - Forks: 11

lkhphuc/lightning-hydra-template Fork of Lightning-AI/deep-learning-project-template

Deep Learning project template best practices with Pytorch Lightning, Hydra, Tensorboard.

Language: Python - Size: 107 KB - Last synced at: over 1 year ago - Pushed at: over 4 years ago - Stars: 151 - Forks: 17

munhouiani/Deep-Packet

Pytorch implementation of deep packet: a novel approach for encrypted traffic classification using deep learning

Language: Jupyter Notebook - Size: 2.8 MB - Last synced at: over 1 year ago - Pushed at: over 2 years ago - Stars: 139 - Forks: 46

drivendataorg/zamba

A Python package for identifying hundreds of kinds of animals, training custom models, and estimating distance from camera trap videos and images

Language: Python - Size: 83.4 MB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 135 - Forks: 31

georgian-io/Knowledge-Distillation-Toolkit 📦

:no_entry: [DEPRECATED] A knowledge distillation toolkit based on PyTorch and PyTorch Lightning.

Language: Python - Size: 76.9 MB - Last synced at: about 1 year ago - Pushed at: over 1 year ago - Stars: 135 - Forks: 25

dahyun-kang/ifsl

[CVPR'22] Official PyTorch implementation of Integrative Few-Shot Learning for Classification and Segmentation

Language: Python - Size: 8.54 MB - Last synced at: 3 months ago - Pushed at: over 1 year ago - Stars: 130 - Forks: 16

dongkyunk/DOLG-pytorch

Unofficial PyTorch Implementation of "DOLG: Single-Stage Image Retrieval with Deep Orthogonal Fusion of Local and Global Features"

Language: Python - Size: 13.7 KB - Last synced at: 12 months ago - Pushed at: over 3 years ago - Stars: 128 - Forks: 15

nmichlo/disent

🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervised, weakly supervised and unsupervised methods ▸ Easily configured and run with Hydra config ▸ Inspired by disentanglement_lib

Language: Python - Size: 18.8 MB - Last synced at: 14 days ago - Pushed at: over 2 years ago - Stars: 127 - Forks: 17