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.

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.

pytorch-tabular

A unified framework for Deep Learning Models on tabular data

ptlflow

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

rl4co

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

pytorch-segmentation-models-trainer

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

pytorch-ie

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

labml

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

torchmocks

Test pytorch code with minimal computational overhead

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

namazu

ML utilities/models for PyTorch, PyTorch Lighting

aiaccel

AIST Toolkit for Accelerating Machine Learning Research

lightning-pose

Accelerated pose estimation and tracking using semi-supervised learning

open-metric-learning

Metric learning and retrieval pipelines, models and zoo.

pycave

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

classy-core

A powerful tool to train and use your classification models.

rul-datasets

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

quickvision

Computer Vision models and training

everyvoice

The EveryVoice TTS Toolkit - Text To Speech for your language

embeddings-for-trees

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