A collection of well established, SOTA models and components.
Visit BoltsPyTorch Lightning Bolts is a community-built deep learning research and production toolbox, featuring a collection of well established and SOTA models and components, pre-trained weights, callbacks, loss functions, data sets, and data modules.
Everything is implemented in Lightning and tested (daily), benchmarked, documented, and works on CPUs, TPUs, GPUs, and 16-bit precision.
What separates Bolts from all the other libraries out there, is that Bolts is built by and used by AI researchers. This means every single Bolt component is modularized so that it can be easily extended or mixed with arbitrary parts of the rest of the code-base.
Bolts has rigorously tested and benchmarked baselines. From VAEs to GANs to GPT to self-supervised models — you don’t have to spend months implementing the baselines to try new ideas. Instead, subclass one of ours and try your idea!
Lightning Bolts includes a collection of non-deep learning algorithms that can train on multiple GPUs and TPUs.
Bolts houses a collection of many of the current state-of-the-art self-supervised algorithms: SimCLR, SwAV, AMDIM, BYOL, CPC-V2, MOCO-V2...
Bolts contains a variety of DQN models you can extend to build your own reinforcement learning models.
The Ultimate Pytorch Research Framework
