🧶 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

✭ 41

python metrics pytorch vae metric-learning datasets representation-learning configurable autoencoders disentanglement disentangled-representations pytorch-lightning

ride

Training wheels, side rails, and helicopter parent for your Deep Learning projects in Pytorch

✭ 18

python Makefile machine-learning framework ai deep-learning pytorch artificial-intelligence pytorch-lightning

gestop

A tool to navigate the desktop with hand gestures. Builds on mediapipe.

✭ 20

python C++Starlark deep-learning pytorch hand-gestures mediapipe pytorch-lightning

pytorch tempest

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

✭ 124

python deep-learning hydra hacktoberfest training-pipeline pytorch-lightning

classy

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

✭ 61

python Jupyter Notebook nlp natural-language-processing deep-learning neural-network transformers pytorch seq2seq sequence-to-sequence natural-language-generation nlp-library bert natural-language-understanding pytorch-lightning bert-fine-tuning

DOLG-pytorch

Unofficial PyTorch Implementation of "DOLG: Single-Stage Image Retrieval with Deep Orthogonal Fusion of Local and Global Features"

✭ 69

python deep-learning cnn pytorch image-retrieval pytorch-lightning dolg

2021-dialogue-summary-competition

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

✭ 86

python Dockerfile nlp dialogue summarization pytorch-lightning huggingface-transformers

lightning-asr

Modular and extensible speech recognition library leveraging pytorch-lightning and hydra.

✭ 36

python hydra speech-recognition asr conformer pytorch-lightning

uetai

Custom ML tracking experiment and debugging tools.

✭ 17

python Dockerfile machine-learning ai data-analysis experiment-tracking model-debugging pytorch-lightning

Neural-HMM

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

✭ 69

Jupyter Notebook python machine-learning text-to-speech deep-learning neural-network pytorch speech-synthesis hidden-markov-model pytorch-lightning

fastface

Light Face Detection using PyTorch Lightning

✭ 71

python pytorch facedetection edge-ai pytorch-lightning

embeddings-for-trees

Set of PyTorch modules for developing and evaluating different algorithms for embedding trees.

✭ 19

python pytorch tree-lstm ml4code pytorch-lightning dgl ml4se embedding-trees

MVSNet pl

MVSNet: Depth Inference for Unstructured Multi-view Stereo using pytorch-lightning

✭ 49

Jupyter Notebook python pytorch multi-view-stereo 3d-reconstruction depth-prediction pytorch-lightning mvsnet

pytorch multi input example

Multi-Input Deep Neural Networks with PyTorch-Lightning - Combine Image and Tabular Data

✭ 40

python tutorial deep-neural-networks tabular-data pytorch pytorch-lightning

quickvision

An Easy To Use PyTorch Computer Vision Library

✭ 49

python Dockerfile computer-vision deep-learning pytorch pytorch-lightning

bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

✭ 56

python nlp transformers lstm theseus pruning quantization bert distillation pytorch-lightning fastbert deebert

lightning-transformers

Flexible components pairing 🤗 Transformers with Pytorch Lightning

✭ 551

python Makefile transformers pytorch hydra pytorch-lightning

weasel

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

✭ 117

python machine-learning deep-learning weak-supervision weakly-supervised-learning end-to-end-learning pytorch-lightning multi-source-weak-supervision

labml

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

map-floodwater-satellite-imagery

This repository focuses on training semantic segmentation models to predict the presence of floodwater for disaster prevention. Models were trained using SageMaker and Colab.

✭ 21

Jupyter Notebook aws hyperparameter-tuning semantic-segmentation sagemaker pytorch-lightning streamlit weights-and-biases

AutoTabular

Automatic machine learning for tabular data. ⚡🔥⚡

✭ 51

python data-science machine-learning deep-learning scikit-learn tabular-data xgboost lightgbm structured-data feature-engineering hpo automl catboost pytorch-lightning

uvadlc notebooks

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

✭ 901

Jupyter Notebook tutorial deep-learning tutorials pytorch flax jax pytorch-lightning optax uvadlc

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.

✭ 555

Jupyter Notebook python pytorch pytorch-lightning pytorch-lightning-template

lightning-hydra-template

PyTorch Lightning + Hydra. A very user-friendly template for rapid and reproducible ML experimentation with best practices. ⚡🔥⚡

deepfillv2-pylightning

Clean minimal implementation of Free-Form Image Inpainting with Gated Convolutions in pytorch lightning. Inspired from pytorch implementation by @avalonstrel.

✭ 13

python deep-learning pytorch image-inpainting pytorch-lightning

BrainMaGe

Brain extraction in presence of abnormalities, using single and multiple MRI modalities

✭ 23

python Jupyter Notebook machine-learning deep-learning pytorch neuroimaging segmentation nih brain-extraction skull-stripping pytorch-lightning

skillful nowcasting

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

✭ 117

python pytorch gan nowcasting pytorch-implementation nowcasting-precipitation pytorch-lightning nowcasting-models

Transformer-QG-on-SQuAD

Implement Question Generator with SOTA pre-trained Language Models (RoBERTa, BERT, GPT, BART, T5, etc.)

✭ 28

python Dockerfile flask text-generation api-server bart albert squad bert question-generation roberta gpt2 pytorch-lightning t5-model

deepaudio-speaker

neural network based speaker embedder

✭ 19

python shell hydra speaker-recognition speaker-verification speaker-diarization pytorch-lightning

VideoTransformer-pytorch

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.

✭ 159

python Jupyter Notebook transformer deeplearning action-recognition pytorch-implmention pretrained-model pytorch-lightning timesformer vivit maskfeat

ConSSL

PyTorch Implementation of SOTA SSL methods

✭ 61

python docker latex imagenet-dataset pytorch-implmention pytorch-ignite cifar-dataset pytorch-lightning

hififace

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

✭ 227

python Dockerfile pytorch face-swapping pytorch-lightning wandb

Fast-AgingGAN

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

✭ 133

python neural-network cnn pytorch generative-adversarial-network gan tensorboard faceapp cycle-gan face-aging pytorch-lightning realtime-face-aging

hydra-zen

Pythonic functions for creating and enhancing Hydra applications

✭ 165

python machine-learning scalable deep-learning configuration pytorch reproducible-science machinelearning configurable reproducibility yaml-configuration dynamic-configuration dataclasses pytorch-lightning

1-39 of 39 pytorch-lightning projects