pytorch-lightning

There are 1137 repositories under pytorch-lightning topic.

  • voicepaw/so-vits-svc-fork

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

    Language:Python9.2k623771.2k

  • XavierXiao/Dreambooth-Stable-Diffusion

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

    Language:Jupyter Notebook7.8k90152806

  • 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:Python5.3k65156459

  • ashleve/lightning-hydra-template

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

    Language:Python5k30249729

  • sktime/pytorch-forecasting

    Time series forecasting with PyTorch

    Language:Python4.6k41916723

  • 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 Notebook3k4296650
  • nerf_pl

    kwea123/nerf_pl

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

    Language:Jupyter Notebook2.8k51210474
  • labml

    labmlai/labml

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

    Language:Python2.3k2750149

  • 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:Python1.7k31521210

  • erikwijmans/Pointnet2_PyTorch

    PyTorch implementation of Pointnet2/Pointnet++

    Language:Python1.7k16155383

  • manujosephv/pytorch_tabular

    A standard framework for modelling Deep Learning Models for tabular data

    Language:Python1.6k18172160

  • 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 Notebook1.5k914193

  • vturrisi/solo-learn

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

    Language:Python1.5k10162196
  • traingenerator

    jrieke/traingenerator

    🧙 A web app to generate template code for machine learning

    Language:Python1.4k3610185

  • 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 Notebook1.3k22142156

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

    Language:Python1.1k1165132

  • 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:Python984150143

  • OML-Team/open-metric-learning

    Metric learning and retrieval pipelines, models and zoo.

    Language:Python9771223372

  • airctic/icevision

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

    Language:Python86722574150

  • 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:Python800168396
  • norse

    norse/norse

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

    Language:Python7691317492

  • ai4co/rl4co

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

    Language:Python71712104127

  • torchmd/torchmd

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

    Language:Python670143184
  • quaterion

    qdrant/quaterion

    Blazing fast framework for fine-tuning similarity learning models

    Language:Python66188147

  • grok-ai/nn-template

    Generic template to bootstrap your PyTorch project.

    Language:Python64813968

  • maum-ai/faceshifter

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

    Language:Python6461630121

  • Lightning-Universe/lightning-transformers

    Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

    Language:Python6122212975

  • mathLab/PINA

    Physics-Informed Neural networks for Advanced modeling

    Language:Python5841620584

  • JeffersonQin/YuzuMarker.FontDetection

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

    Language:Python51733425

  • krasserm/perceiver-io

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

    Language:Python49693244

  • changjonathanc/minLoRA

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

    Language:Jupyter Notebook4829928

  • hmorimitsu/ptlflow

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

    Language:Python449114351
  • 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.

    Language:Python43855260

  • mit-ll-responsible-ai/hydra-zen

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

    Language:Python424514519
  • machin

    iffiX/machin

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

    Language:Python41931851

  • zjukg/NeuralKG

    [Tool] For Knowledge Graph Representation Learning

    Language:Python388104871