Awesome Open Source

Programming Languages

Search results for pytorch lightning

pytorch-lightning x

288 search results found

So Vits Svc Fork ⭐ 8,080

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

Dreambooth Stable Diffusion ⭐ 7,184

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

Lightning Hydra Template ⭐ 3,680

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

Pytorch Forecasting ⭐ 3,439

Time series forecasting with PyTorch

Nerf_pl ⭐ 2,291

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

Uvadlc_notebooks ⭐ 1,860

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

Labml ⭐ 1,771

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

Lightning Flash ⭐ 1,729

Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains

Solo Learn ⭐ 1,318

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

Traingenerator ⭐ 1,184

🧙 A web app to generate template code for machine learning

Pytorch_tabular ⭐ 1,138

A standard framework for modelling Deep Learning Models for tabular data

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

Pointnet2_pytorch ⭐ 920

PyTorch implementation of Pointnet2/Pointnet++

Pytorch Lightning Template ⭐ 878

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.

Icevision ⭐ 823

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

Detoxify ⭐ 774

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].

Open Metric Learning ⭐ 699

Library for metric learning pipelines and models.

Nn Template ⭐ 614

Generic template to bootstrap your PyTorch project.

Lightning Transformers ⭐ 608

Flexible components pairing 🤗 Transformers with ⚡ Pytorch Lightning

Quaterion ⭐ 580

Blazing fast framework for fine-tuning similarity learning models

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

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

Torchmd ⭐ 465

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

Faceshifter ⭐ 384

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

Perceiver Io ⭐ 382

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

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.

Transformersum ⭐ 362

Models to perform neural summarization (extractive and abstractive) using machine learning transformers and a tool to convert abstractive summarization datasets to the extractive task.

Yuzumarker.fontdetection ⭐ 361

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

PyTorch (and PyTorch Lightning) implementation of Neural Style Transfer, Pix2Pix, CycleGAN, and Deep Dream!

Scenerf ⭐ 284

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

Library for automatic retraining and continual learning

Neuralkg ⭐ 263

[Tool] For Knowledge Graph Representation Learning

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

Hydra Zen ⭐ 252

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

Casmvsnet_pl ⭐ 246

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

Stock Prediction Neural Network And Machine Learning Examples ⭐ 246

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

Minlora ⭐ 217

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

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

Lightning Pose ⭐ 202

Accelerated pose estimation and tracking using semi-supervised convolutional networks.

Pytorch_tempest ⭐ 193

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

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

Fast Aginggan ⭐ 179

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

Skillful_nowcasting ⭐ 171

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

Nsff_pl ⭐ 167

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

Sinnerf ⭐ 162

[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

Multi Camera Person Re Identification ⭐ 154

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

Openkiwi ⭐ 145

Open-Source Machine Translation Quality Estimation in PyTorch

Neural Hmm ⭐ 143

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

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

Yet Another Lightning Hydra Template ⭐ 129

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

Net2net ⭐ 128

Network-to-Network Translation with Conditional Invertible Neural Networks

Fusilli ⭐ 120

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 🌸

Openglue ⭐ 118

Open Source Graph Neural Net Based Pipeline for Image Matching

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

Distiller ⭐ 112

A large scale study of Knowledge Distillation.

Traditional Machine Learning Models for Large-Scale Datasets in PyTorch.

Official implementation for ICDAR 2021 best poster paper "Handwritten Mathematical Expression Recognition with Bidirectionally Trained Transformer"

Covid Mask Detector ⭐ 102

Detect whether a person is wearing a mask or not

L Verse ⭐ 100

L-Verse: Bidirectional Generation Between Image and Text

Deep Packet ⭐ 100

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

Matsciml ⭐ 94

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.

2021 Dialogue Summary Competition ⭐ 92

[2021 훈민정음 한국어 음성•자연어 인공지능 경진대회] 대화요약 부문 알라꿍달라꿍 팀의 대화요약 학습 및 추론 코드를 공유하기 위한 레포입니다.

Pytorch Lightning Gans ⭐ 91

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

Videotransformer Pytorch ⭐ 90

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.

Consistency Models ⭐ 89

Toolkits for OpenAI's Consistency Models.

🧶 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

Feartracker ⭐ 85

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

Fastface ⭐ 83

Light Face Detection using PyTorch Lightning

classy is a simple-to-use library for building high-performance Machine Learning models in NLP.

KUIELAB-MDX-Net got the 2nd place on the Leaderboard A and the 3rd place on the Leaderboard B in the MDX-Challenge ISMIR 2021

Bert Squeeze ⭐ 77

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

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

Lightningfsl ⭐ 76

LightningFSL: Pytorch-Lightning implementations of Few-Shot Learning models.

Monodepth_benchmark ⭐ 74

Code for "Deconstructing Monocular Depth Reconstruction: The Design Decisions that Matter" (https://arxiv.org/abs/2208.01489)

Whos There ⭐ 73

The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends

Conditioned Source Separation Lasaft ⭐ 73

A PyTorch implementation of the paper: "LaSAFT: Latent Source Attentive Frequency Transformation for Conditioned Source Separation" (ICASSP 2021)

A training and inference framework for open ner and re models! 信息抽取模型的统一训练和推理框架，包含丰富的开源SOTA模型

Energizer ⭐ 71

An active learning library for Pytorch based on Pytorch-Lightning.

Knowledge Distillation Toolkit ⭐ 71

A knowledge distillation toolkit based on PyTorch and PyTorch Lightning.

Hififace ⭐ 70

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

Research code for paper "Frido: Feature Pyramid Diffusion for Complex Scene Image Synthesis"

Production-ready audio and video transcription app that can run on your laptop or in the cloud.

Satellighte ⭐ 68

📡 PyTorch Lightning Implementations of Recent Satellite Image Classification !

Wheat Detection challenge on Kaggle

Gpumonitor ⭐ 65

TF 2.x and PyTorch Lightning Callbacks for GPU monitoring

Time series forecasting with PyTorch

Torchline ⭐ 63

Easy to use Pytorch

Pytorch Ie ⭐ 63

PyTorch-IE: State-of-the-art Information Extraction in PyTorch

Comma10k Baseline ⭐ 59

A baseline segmentation example using the comma10k dataset (WIP)

Dyffusion ⭐ 58

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

Full_stack_transformer ⭐ 57

Pytorch library for end-to-end transformer models training, inference and serving

Studio Lab ⭐ 57

Studio Lab is hackable boilerplate for PyTorch Lightning driven deep learning research in Lightning AI Studios

Code2seq ⭐ 54

PyTorch's implementation of the code2seq model.

The official repo: "McNet: Fuse Multiple Cues for Multichannel Speech Enhancement" submitted to ICASSP 2023

Head Segmentation ⭐ 52

👦 Human head semantic segmentation

Multi_part_assembly ⭐ 51

Code Accompany NeurIPS 2022 Dataset Paper: Breaking Bad. A Shape Assembly Benchmark Implemented in PyTorch

Quickvision ⭐ 51

An Easy To Use PyTorch Computer Vision Library

Alias Free Gan ⭐ 50

Unofficial Alias-Free GAN implementation. Based on rosinality's version with expanded training and inference options.

Audiossl ⭐ 50

A library built for easier audio self-supervised training, downstream tasks evaluation

Finetuning Scheduler ⭐ 49

A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.

1-100 of 288 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.