GitHub topics: pytorch-lightning

Repositories

dsgoficial/pytorch_segmentation_models_trainer

Framework to train semantic segmentation models on Pytorch using yaml config files

Language: Python - Size: 46.1 MB - Last synced at: about 20 hours ago - Pushed at: about 21 hours ago - Stars: 17 - Forks: 1

twsl/pytorchcocotools

Unofficial APIs for the MS-COCO dataset using PyTorch

Language: Python - Size: 2.36 MB - Last synced at: about 23 hours ago - Pushed at: 1 day ago - Stars: 2 - Forks: 0

DavidZhang73/pytorch-lightning-template

A template for simple deep learning projects using Lightning

Language: Python - Size: 50.8 KB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 25 - Forks: 1

bshtmichielsen/clothing_sorter

Sorting images of attire using classification.

Language: Jupyter Notebook - Size: 102 MB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 1 - Forks: 0

bshtmichielsen/house_price_predictor

Predicting the prices of houses using regression.

Language: Jupyter Notebook - Size: 804 KB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 1 - Forks: 0

speediedan/finetuning-scheduler

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

Language: Python - Size: 2.76 MB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 66 - Forks: 7

copev313/Reveal-AI

A study of SOTA models and their ability to detect AI-generated imagery when faced with common transformations like cropping and resizing.

Size: 151 KB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 0 - Forks: 0

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: 1 day ago - Pushed at: 1 day ago - Stars: 1,527 - Forks: 196

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: 4 days ago - Pushed at: 4 days ago - Stars: 204 - Forks: 22

alephpi/Texo

A minimalist SOTA LaTeX OCR model which contains only 20M parameters and runs in browser. | 超轻量SOTA LaTeX公式识别模型，20M参数量，可在浏览器中运行

Language: Python - Size: 4.25 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 135 - Forks: 3

voicepaw/so-vits-svc-fork

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

Language: Python - Size: 21.1 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 9,175 - Forks: 1,221

jyoti2000-star/CASM

📦 Simplify application management with CASM, your solution for streamlined deployment and orchestration across different environments.

Language: Python - Size: 487 KB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 0 - Forks: 0

heitornolla/Computer-Vision-Studies

A lot of useful books, lessons and codes I've studied. All neatly compiled, hoping to save time for other learners :)

Language: Jupyter Notebook - Size: 1.74 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 0 - Forks: 0

twsl/lit-mlflow

An improved Lightning mlflow logger to replace mlflow.autolog()

Language: Python - Size: 940 KB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 0 - Forks: 0

sktime/pytorch-forecasting

Time series forecasting with PyTorch

Language: Python - Size: 45.2 MB - Last synced at: 7 days ago - Pushed at: 8 days ago - Stars: 4,640 - Forks: 722

aistairc/aiaccel

AIST Toolkit for Accelerating Machine Learning Research

Language: Python - Size: 4.56 MB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 34 - Forks: 4

wHunta/Custom-image-generator-from-0-to-9

🎨 Create custom digit images using a Conditional GAN trained on the MNIST dataset, all within a clean PyTorch Lightning framework.

Language: Jupyter Notebook - Size: 1.36 MB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 0 - Forks: 0

giacomoguidotto/pinn Fork of johnthagen/python-blueprint

🤖 A brand new library...

Language: Python - Size: 2.44 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 0 - Forks: 0

mathLab/PINA

Physics-Informed Neural networks for Advanced modeling

Language: Python - Size: 135 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 583 - Forks: 84

msuvorov7/proslushka

ASR model

Language: Python - Size: 84 KB - Last synced at: 9 days ago - Pushed at: 10 days ago - Stars: 0 - Forks: 0

openclimatefix/skillful_nowcasting

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

Language: Python - Size: 211 KB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 271 - Forks: 64

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: 9 days ago - Stars: 1,588 - Forks: 160

aymuos15/nnBenchmark

The last benchmark library you will need for common medical image analysis tasks

Language: Python - Size: 596 KB - Last synced at: 11 days ago - Pushed at: 11 days ago - Stars: 0 - Forks: 0

bhimrazy/minimal-ddp-example

Minimal DDP Example with LitData and PyTorch Lightning

Language: Python - Size: 3.91 KB - Last synced at: 11 days ago - Pushed at: 11 days ago - Stars: 1 - Forks: 0

mit-ll-responsible-ai/hydra-zen

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

Language: Python - Size: 23.9 MB - Last synced at: 12 days ago - Pushed at: 12 days ago - Stars: 419 - Forks: 19

jrieke/traingenerator

🧙 A web app to generate template code for machine learning

Language: Python - Size: 9.86 MB - Last synced at: about 5 hours ago - Pushed at: about 2 years ago - Stars: 1,379 - Forks: 185

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: 2 days ago - Pushed at: about 2 years ago - Stars: 1,738 - Forks: 210

grok-ai/nn-template

Generic template to bootstrap your PyTorch project.

Language: Python - Size: 2.68 MB - Last synced at: 11 days ago - Pushed at: about 2 years ago - Stars: 648 - Forks: 68

EveryVoiceTTS/EveryVoice

The EveryVoice TTS Toolkit - Text To Speech for your language

Language: Python - Size: 10.7 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 41 - Forks: 2

Kor4yz/Recognition-of-house-numbers-for-the-delivery-robot-using-a-neural-network.

Распознавание номеров домов (SVHN) для робота-курьера: чистый PyTorch/Lightning конвейер, воспроизводимые эксперименты, Docker, CI, Streamlit-демо, модельная карточка, тесты.

Language: Jupyter Notebook - Size: 2.67 MB - Last synced at: 17 days ago - Pushed at: 17 days ago - Stars: 0 - Forks: 0

junhsss/consistency-models

A Toolkit for OpenAI's Consistency Models.

Language: Python - Size: 55.1 MB - Last synced at: 16 days ago - Pushed at: over 2 years ago - Stars: 206 - Forks: 12

XReflection is a neat toolbox tailored for single-image reflection removal(SIRR). We offer state-of-the-art SIRR solutions for training and inference, with a high-performance data pipeline, multi-GPU/TPU/NPU support, and more!

Language: Python - Size: 4.36 MB - Last synced at: 17 days ago - Pushed at: 17 days ago - Stars: 22 - Forks: 5

labmlai/labml

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

Language: Python - Size: 101 MB - Last synced at: 17 days ago - Pushed at: 7 months ago - Stars: 2,249 - Forks: 148

adosar/aidsorb

Python package for deep learning on molecular point clouds.

Language: Python - Size: 16.6 MB - Last synced at: 18 days ago - Pushed at: 18 days ago - Stars: 6 - Forks: 2

project-lighter/lighter

Streamline deep learning experiments using config files

Language: Python - Size: 6.9 MB - Last synced at: about 7 hours ago - Pushed at: about 8 hours ago - Stars: 68 - Forks: 7

zuoxingdong/rlbidder

Reinforcement learning auto-bidding library for research and production.

Language: Python - Size: 1.56 MB - Last synced at: 19 days ago - Pushed at: 20 days ago - Stars: 1 - Forks: 0

THAMIZH-ARASU/3D-SEGMENTATION-OF_MEDICAL_IMAGES

A modular, extensible, and production-ready framework for 3D CT scan segmentation using PyTorch Lightning, TorchIO, and modern deep learning best practices which supports preprocessing, training, prediction, and evaluation for medical image segmentation tasks..

Language: Python - Size: 654 KB - Last synced at: 20 days ago - Pushed at: 20 days ago - Stars: 0 - Forks: 0

torchmd/torchmd

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

Language: Python - Size: 2.51 MB - Last synced at: 19 days ago - Pushed at: 10 months ago - Stars: 664 - Forks: 84

borhanMorphy/fastface

Light Face Detection using PyTorch Lightning

Language: Python - Size: 1.03 MB - Last synced at: 15 days ago - Pushed at: about 2 years ago - Stars: 82 - Forks: 10

aadityaguptaaa/diabetic-retinopathy-detection

About Diabetic Retinopathy Detection: Utilizing Multiprocessing for Processing Large Datasets and Transfer Learning to Fine-Tune Deep Learning Models | PyTorch

Language: Python - Size: 1.56 GB - Last synced at: 21 days ago - Pushed at: 21 days ago - Stars: 4 - Forks: 1

Moenupa/LyCIL Fork of SmallPigPeppa/lightning_cil

LyCIL: A Pytorch-Lightning 2.x Toolbox for Continual/Incremental Learning. Inspired by https://github.com/LAMDA-CL/PyCIL.

Language: Python - Size: 600 KB - Last synced at: 21 days ago - Pushed at: 21 days ago - Stars: 0 - Forks: 0

neuro-ml/thunder

🌩️ The Deep Learning framework based on Lightning

Language: Python - Size: 3.69 MB - Last synced at: 14 days ago - Pushed at: 2 months ago - Stars: 11 - Forks: 1

paninski-lab/lightning-pose

Accelerated pose estimation and tracking using semi-supervised learning

Language: Python - Size: 144 MB - Last synced at: 22 days ago - Pushed at: 22 days ago - Stars: 274 - Forks: 44

Dexterp37/fontina

Data generation, model training and inference for Visual Font Recognition using PyTorch

Language: Python - Size: 56.6 KB - Last synced at: 8 days ago - Pushed at: almost 2 years ago - Stars: 16 - Forks: 2

yoyolicoris/music-spectrogram-diffusion-pytorch

Language: Python - Size: 288 KB - Last synced at: 20 days ago - Pushed at: almost 3 years ago - Stars: 85 - Forks: 6

zeroone-universe/RealTimeBWE

Unofficial Pytorch Lightning Implementation of "Real-time Speech Frequency Bandwidth Extension"

Language: Python - Size: 1.54 MB - Last synced at: 23 days ago - Pushed at: 23 days ago - Stars: 36 - Forks: 5

CANARI-ML/canari-ml

Main repo tracking/including early CANARI-ML work

Language: Python - Size: 1.03 MB - Last synced at: 23 days ago - Pushed at: 23 days ago - Stars: 7 - Forks: 0

lebellig/flow-matching

Annotated Flow Matching paper

Language: Jupyter Notebook - Size: 290 KB - Last synced at: 23 days ago - Pushed at: about 1 year ago - Stars: 214 - Forks: 12

3dlg-hcvc/multi3drefer

[ICCV 2023] Multi3DRefer: Grounding Text Description to Multiple 3D Objects

Language: Python - Size: 7.18 MB - Last synced at: 24 days ago - Pushed at: 24 days ago - Stars: 92 - Forks: 4

Waidhoferj/dance-classifier

Deep learning model which classifies the dance style that best accompanies a provided song

Language: Python - Size: 146 KB - Last synced at: 24 days ago - Pushed at: 24 days ago - Stars: 1 - Forks: 1

p2pfl/p2pfl

P2PFL is a decentralized federated learning library that enables federated learning on peer-to-peer networks using gossip protocols, making collaborative AI model training possible without reliance on central servers.

Language: Jupyter Notebook - Size: 59.9 MB - Last synced at: 24 days ago - Pushed at: about 1 month ago - Stars: 120 - Forks: 31

DyeDye-diyung/Framework_for_Financial_Time-Series_Forecasting

This repository contains the official source code and data for the research paper: Enhancing stock price forecasting with a modular deep learning framework incorporating plug-and-play Transformer variants.

Language: Python - Size: 381 MB - Last synced at: 26 days ago - Pushed at: 27 days ago - Stars: 2 - Forks: 0

seucoin/unKR

A Python Library for Uncertain Knowledge Graph Reasoning

Language: Python - Size: 142 MB - Last synced at: 27 days ago - Pushed at: 28 days ago - Stars: 115 - Forks: 72

AtticusZeller/Pytorch-Lightning-uv

📦 Zero-config Deep Learning template with PyTorch Lightning, UV package manager, W&B tracking, and modern Python tooling 🚀

Language: Python - Size: 4.29 MB - Last synced at: 28 days ago - Pushed at: 28 days ago - Stars: 3 - Forks: 0

jhauret/vibravox

Speech to Phoneme, Bandwidth Extension and Speaker Verification using the Vibravox dataset.

Language: Python - Size: 2.29 MB - Last synced at: 16 days ago - Pushed at: 5 months ago - Stars: 42 - Forks: 5

tshu-w/lightning-template

PyTorch Lightning template for rapid, scalable and reproducible ML experimentation with best practices.

Language: Python - Size: 146 KB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 25 - Forks: 2

veritable-tech/pytorch-lightning-spells

Some useful plugins for pytorch-lightning

Language: Python - Size: 235 KB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 9 - Forks: 2

saeedmehrang/cvmt

Bone Age Maturity estimation using a Lateral Cephalogram X-ray image and deep neural networks (UNet)

Language: Jupyter Notebook - Size: 77.2 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 1 - Forks: 0

wiktorlazarski/ai-awesome-project-template

🤖 Project template for your next awesome AI project. 🦾

Language: Python - Size: 20.5 KB - Last synced at: 1 day ago - Pushed at: over 3 years ago - Stars: 25 - Forks: 3

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: Python - Size: 50.8 MB - Last synced at: about 1 month ago - Pushed at: 4 months ago - Stars: 1,119 - Forks: 132

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: 27 days ago - Pushed at: over 2 years ago - Stars: 127 - Forks: 17

tilman151/rul-datasets

A collection of datasets for RUL estimation as Lightning Data Modules.

Language: Python - Size: 1.13 MB - Last synced at: 13 days ago - Pushed at: over 1 year ago - Stars: 48 - Forks: 2

borchero/pycave

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

Language: Python - Size: 683 KB - Last synced at: 23 days ago - Pushed at: 23 days ago - Stars: 128 - Forks: 14

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 1 month ago - Pushed at: about 1 year ago - Stars: 4,900 - Forks: 712

Borda/kaggle_image-classify 📦

Various Kaggle image classification challenges solutions

Language: Jupyter Notebook - Size: 12 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 41 - Forks: 12

lus105/DeepVisionXplain

Versatile model training environment with XAI included

Language: Python - Size: 15.3 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 5 - Forks: 0

vishwajitsarnobat/INDRA-Sat-Diff

A flexible PyTorch Lightning framework for climate forecasting, featuring a customizable, knowledge-aligned latent diffusion model for high-resolution predictions.

Language: Python - Size: 380 KB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 0 - Forks: 0

DivyenduDutta/NanoDefectNet

End-to-end deep learning app for detecting defects in semiconductor wafermaps

Language: Python - Size: 103 KB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 0 - Forks: 0

Pacatro/UcoRecSys

Application of Recommendation Systems in Educational Environments

Language: Jupyter Notebook - Size: 26.7 MB - Last synced at: 17 days ago - Pushed at: about 2 months ago - Stars: 1 - Forks: 0

Luciennnnnnn/liif-lightning-hydra

An reimplement of liif(Learning Continuous Image Representation with Local Implicit Image Function) using lightning+hydra

Language: Python - Size: 69.3 KB - Last synced at: 15 days ago - Pushed at: over 4 years ago - Stars: 11 - Forks: 1

lebellig/discrete-fm

Educational implementation of the Discrete Flow Matching paper

Language: Jupyter Notebook - Size: 69.3 MB - Last synced at: about 1 month ago - Pushed at: about 1 year ago - Stars: 117 - Forks: 7

ArneBinder/pytorch-ie

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

Language: Python - Size: 1.86 MB - Last synced at: 13 days ago - Pushed at: about 2 months ago - Stars: 77 - Forks: 8

Audio-WestlakeU/audiossl

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

Language: Python - Size: 13 MB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 130 - Forks: 10

zjukg/NeuralKG

[Tool] For Knowledge Graph Representation Learning

Language: Python - Size: 62.8 MB - Last synced at: about 1 month ago - Pushed at: over 1 year ago - Stars: 381 - Forks: 70

norse/norse

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

Language: Python - Size: 110 MB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 759 - Forks: 93

monajemi-arman/dental_decay_detection

Detection of dental carries using deep learning model

Language: Python - Size: 340 KB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 0 - Forks: 0

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: about 2 months ago - Pushed at: almost 2 years ago - Stars: 490 - Forks: 44

3dlg-hcvc/minsu3d

MINSU3D: MinkowskiEngine-powered Scene Understanding in 3D

Language: Python - Size: 20.3 MB - Last synced at: about 1 month ago - Pushed at: over 1 year ago - Stars: 42 - Forks: 5

hmorimitsu/ptlflow

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

Language: Python - Size: 9.62 MB - Last synced at: about 2 months ago - Pushed at: 2 months ago - Stars: 435 - Forks: 50

SerezD/mi_ml_gen

Code for Paper: "A Mutual Information Perspective on Multiple Latent Variable Generative Models for Positive View Generation"

Language: Python - Size: 592 KB - Last synced at: about 1 month ago - Pushed at: about 2 months ago - Stars: 0 - Forks: 0

krish567366/automl_self_improvement

A next-gen toolkit for autonomous machine learning that automatically optimizes models using reinforcement learning, evolutionary algorithms, and generative AI. Combines meta-learning for cross-task adaptation, hybrid hyperparameter tuning, and secure MLOps to accelerate AI deployment.

Language: Python - Size: 33.2 KB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 1 - Forks: 0