PyTorch Lightning Spells

Some useful plugins for PyTorch Lightning.

IMPORTANT NOTICE: There is ongoing work to make this package fully compatible with PyTorch Lightning version 2.1. Submitting issues and pull requests for compatibility with this version is highly appreciated.

Support for older versions of PyTorch Lightning:

Please install version 0.1.0 if you are working with PyTorch Lightning version 2.1.
Please install version 0.0.5 if you are working with PyTorch Lightning version 1.3 or 1.4.

Installation

pip install pytorch-lightning-spells

or the latest in the main branch:

pip install https://github.com/veritable-tech/pytorch-lightning-spells/archive/master.zip

Package Rankings

Top 18.59% on Pypi.org

Badges

Extracted from project README

Related Projects

torchmetrics

Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.

22 Dec 2020 1,998

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

31 Mar 2019 27,777

ffcv_pytorch_lightning

[FFCV-PL] manage fast data loading with ffcv and pytorch lightning

06 Dec 2022 13

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

11 Mar 2020 2,359

ebtorch

🤓🔥 Collection of PyTorch additions, extensions, utilities, uses and abuses

19 Feb 2021 6

torchx

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iter...

04 May 2021 329

lightning-bolts

Toolbox of models, callbacks, and datasets for AI/ML researchers.

25 Mar 2020 1,645

jellyml

JellyML is an open-source tool (python API and command line) for effortlessly embedding a snapsho...

01 Feb 2023 10

rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

01 Feb 2022 1,823

plkit

A wrapper of pytorch-lightning that makes you write even less code.

20 Jul 2020 5

data

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.

12 May 2021 1,071

light-the-torch

Install PyTorch distributions with computation backend auto-detection

09 Jul 2020 218

torchility

A pytorch-lightning based tool for training pytorch model more simply

14 May 2021 19

torchgeo

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

21 May 2021 2,232

tox-ltt

Install PyTorch distributions with light-the-torch

13 Jul 2020 4