Bolts package provides a variety of components to extend PyTorch Lightning, such as callbacks & datasets, for applied research and production. Torch ORT converts your model into an optimized ONNX graph, speeding up training & inference when using NVIDIA or AMD GPUs. We can introduce sparsity during fine-tuning with SparseML, which ultimately allows us to leverage the DeepSparse engine to see performance improvements at inference time.
Features
- Accelerate Lightning Training with the Torch ORT Callback
- Documentation available
- Examples available
- Deep Learning components for extending PyTorch Lightning
- Torch ORT converts your model into an optimized ONNX graph
- Introduce Sparsity with the SparseMLCallback to Accelerate Inference
Categories
Machine LearningLicense
Apache License V2.0Follow Lightning Bolts
You Might Also Like
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of Lightning Bolts!