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.

ptlflow

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

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.

open-metric-learning

Metric learning and retrieval pipelines, models and zoo.

pytorch-segmentation-models-trainer

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

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

aiaccel

AIST Toolkit for Accelerating Machine Learning Research

finetuning-scheduler

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

lightning-pose

Accelerated pose estimation and tracking using semi-supervised learning

pina-mathlab

Physics-Informed Neural networks for Advanced modeling

rl4co

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

embeddings-for-trees

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

lightning-transformers

Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

rul-datasets

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

mushin-py

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

pycave

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

zamba

A Python package for identifying hundreds of kinds of animals, training custom models, and estimating distance from camera trap videos and images

everyvoice

The EveryVoice TTS Toolkit - Text To Speech for your language

classifier-trains

This repository aims to provide a primitive tool to finetune state-of-the-art models with PyTorch implementation, similar to Nvidia TAO but with more flexibility in augmentation and models.

namazu

NaMAZU: Pretty Usefull Library