pytorch-lightning

There are 1060 repositories under pytorch-lightning topic.

  • voicepaw/so-vits-svc-fork

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

    Language:Python8.9k663691.2k

  • XavierXiao/Dreambooth-Stable-Diffusion

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

    Language:Jupyter Notebook7.7k91151802

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

  • ashleve/lightning-hydra-template

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

    Language:Python4.5k30249683

  • sktime/pytorch-forecasting

    Time series forecasting with PyTorch

    Language:Python4.2k44831668
  • nerf_pl

    kwea123/nerf_pl

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

    Language:Jupyter Notebook2.8k51210481

  • 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 Notebook2.8k4193617
  • labml

    labmlai/labml

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

    Language:Python2.1k2750137

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

  • erikwijmans/Pointnet2_PyTorch

    PyTorch implementation of Pointnet2/Pointnet++

    Language:Python1.6k16152359

  • manujosephv/pytorch_tabular

    A standard framework for modelling Deep Learning Models for tabular data

    Language:Python1.5k19163151

  • vturrisi/solo-learn

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

    Language:Python1.5k11162190

  • 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.4k913197
  • 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.3k23142158

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

  • OML-Team/open-metric-learning

    Metric learning and retrieval pipelines, models and zoo.

    Language:Python9171123168

  • airctic/icevision

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

    Language:Python85722574148

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

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

    norse/norse

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

    Language:Python7131416987
  • quaterion

    qdrant/quaterion

    Blazing fast framework for fine-tuning similarity learning models

    Language:Python65798145

  • grok-ai/nn-template

    Generic template to bootstrap your PyTorch project.

    Language:Python64113967

  • maum-ai/faceshifter

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

    Language:Python6301828120

  • Lightning-Universe/lightning-transformers

    Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

    Language:Python6092212976

  • torchmd/torchmd

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

    Language:Python605143177

  • ai4co/rl4co

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

    Language:Python53989895

  • JeffersonQin/YuzuMarker.FontDetection

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

    Language:Python47633022

  • krasserm/perceiver-io

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

    Language:Python45693241

  • cccntu/minLoRA

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

    Language:Jupyter Notebook45410931
  • 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:Python42955258
  • machin

    iffiX/machin

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

    Language:Python40751851

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

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

    Language:Python367614216

  • zjukg/NeuralKG

    [Tool] For Knowledge Graph Representation Learning

    Language:Python366104766

  • astra-vision/SceneRF

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

    Language:Python36563833

  • maum-ai/hififace

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

    Language:Python362241859