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.

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.

so-vits-svc-fork

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

ptlflow

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

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

rl4co

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

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

finetuning-scheduler

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

pina-mathlab

Physics-Informed Neural networks for Advanced modeling

pytorch-segmentation-models-trainer

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

aiaccel

AIST Toolkit for Accelerating Machine Learning Research

namazu

ML utilities/models for PyTorch, PyTorch Lighting

open-metric-learning

OML is a PyTorch-based framework to train and validate the models producing high-quality embeddings.

classy-core

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

torchmocks

Test pytorch code with minimal computational overhead

rul-datasets

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

pycave

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

lightning-pose

Accelerated pose estimation and tracking using semi-supervised learning

ranzen

A python toolkit facilitating machine-learning experimentation.

quadra

Quadra: Effortless and reproducible deep learning workflows with configuration files.

everyvoice

The EveryVoice TTS Toolkit - Text To Speech for your language