Pytorch Lightning Python Packages

hydra-zen

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

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.

epokio

Phone alerts when an ML training run stalls, hits NaN or finishes, from the logs it already writes. No code changes, no account.

ptlflow

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

finetuning-scheduler

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

torch-pointcloud

PyTorch Point Cloud models, scripts, pretrained weights -- PointNet++, DGCNN, KPConv, RandLA-Net, VoteNet, SPVCNN, 3DETR, PointTransformer, PointMamba and more

so-vits-svc-fork

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

pytorch-tabular

A unified framework for Deep Learning Models on tabular data

anypinn

🤖 A modular and flexible solution to build Physics-Informed Neural Networks (PINNs) for any mathematical problem.

labml

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

pytorch-ie

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

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

aiaccel

AIST Toolkit for Accelerating Machine Learning Research

denoisers

Simple PyTorch Denoisers for Waveform Audio

rl4co

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

lightning-pose

Accelerated 2D and 3D pose estimation using semi-supervised learning

namazu

ML utilities/models for PyTorch, PyTorch Lighting

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

classy-core

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

everyvoice

The EveryVoice TTS Toolkit - Text To Speech for your language

open-metric-learning

Metric learning and retrieval pipelines, models and zoo.

ckpt-chaos

Crash-consistency testing for ML training checkpoints: kill a save at every file boundary, resume, classify the outcome

mushin-py

Boilerplate-free, reproducible ML experiment workflows: run a sweep, get a labeled xarray dataset back.

ranzen

A python toolkit facilitating machine-learning experimentation.