Some useful plugins for pytorch-lightning
APACHE-2.0 License
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iter...
JellyML is an open-source tool (python API and command line) for effortlessly embedding a snapsho...
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
A pytorch-lightning based tool for training pytorch model more simply
Install PyTorch distributions with light-the-torch
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data
A wrapper of pytorch-lightning that makes you write even less code.
Install PyTorch distributions with computation backend auto-detection
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
[FFCV-PL] manage fast data loading with ffcv and pytorch lightning
Toolbox of models, callbacks, and datasets for AI/ML researchers.
🤓🔥 Collection of PyTorch additions, extensions, utilities, uses and abuses