pytorch-lightning

There are 1122 repositories under pytorch-lightning topic.

  • voicepaw/so-vits-svc-fork

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

    Language:Python9.1k643741.2k

  • XavierXiao/Dreambooth-Stable-Diffusion

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

    Language:Jupyter Notebook7.7k90152806

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

  • ashleve/lightning-hydra-template

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

    Language:Python4.9k30249709

  • sktime/pytorch-forecasting

    Time series forecasting with PyTorch

    Language:Python4.5k43910704

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

    kwea123/nerf_pl

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

    Language:Jupyter Notebook2.8k53210477
  • labml

    labmlai/labml

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

    Language:Python2.2k2750146

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

  • manujosephv/pytorch_tabular

    A standard framework for modelling Deep Learning Models for tabular data

    Language:Python1.6k18165153

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

  • vturrisi/solo-learn

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

    Language:Python1.5k11162192
  • traingenerator

    jrieke/traingenerator

    🧙 A web app to generate template code for machine learning

    Language:Python1.4k3610186

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

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

  • OML-Team/open-metric-learning

    Metric learning and retrieval pipelines, models and zoo.

    Language:Python9721323371

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

  • airctic/icevision

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

    Language:Python86622574149

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

    norse/norse

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

    Language:Python7581617393

  • ai4co/rl4co

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

    Language:Python6728100120

  • torchmd/torchmd

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

    Language:Python658143179
  • 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:Python64613968

  • maum-ai/faceshifter

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

    Language:Python6451730121

  • Lightning-Universe/lightning-transformers

    Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

    Language:Python6122212975

  • mathLab/PINA

    Physics-Informed Neural networks for Advanced modeling

    Language:Python5661316181

  • JeffersonQin/YuzuMarker.FontDetection

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

    Language:Python50733024

  • krasserm/perceiver-io

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

    Language:Python48793243

  • changjonathanc/minLoRA

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

    Language:Jupyter Notebook474
  • 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:Python43955260

  • hmorimitsu/ptlflow

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

    Language:Python42893843
  • machin

    iffiX/machin

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

    Language:Python41651851

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

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

    Language:Python406614217

  • zjukg/NeuralKG

    [Tool] For Knowledge Graph Representation Learning

    Language:Python380104870