lightning-bolts

Deep learning toolkit

Provides a toolbox of components to extend PyTorch Lightning for deep learning research and production

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

GitHub

2k stars
23 watching
323 forks
Language: Python
last commit: 19 days ago
Linked from 1 awesome list

aiganimage-processingmachine-learningnatural-language-processingpytorchsupervised-learning

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,217
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,666
eduardoleao052/js-pytorch A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. 1,084
ne7ermore/torch-light A comprehensive PyTorch-based deep learning repository with examples and implementations of various models and techniques. 535
dmarnerides/pydlt A PyTorch-based toolbox for building and training deep learning models with ease. 204
kasnerz/lightnlg A minimalistic codebase for training and interacting with NLG models from HuggingFace Transformers using PyTorch Lightning. 3
dmmiller612/sparktorch A PyTorch implementation on Apache Spark for distributed deep learning model training and inference. 339
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432
bloodaxe/pytorch-toolbelt A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. 1,520
tristandeleu/pytorch-meta Provides tools and datasets for meta-learning and few-shot learning in deep learning 1,987
torch/cutorch Provides a CUDA backend for the PyTorch deep learning framework 336
norse/norse Deep learning framework for spiking neural networks 673
magnet-dl/magnet An API that simplifies the development of deep learning architectures by providing a high-level abstraction around PyTorch. 360
rdspring1/pytorch_gbw_lm Trains a large-scale PyTorch language model on the 1-Billion Word dataset 123