torchfold

Batching library

A PyTorch module for dynamic batching and optimized computation on deep neural networks

Tools for PyTorch

GitHub

221 stars
17 watching
28 forks
Language: Python
last commit: over 2 years ago
deep-learningdeep-neural-networksmachine-learningpytorch

Related projects:

Repository Description Stars
vermeille/torchelie A collection of utility functions and tools for building deep learning models with PyTorch 111
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
torch/cutorch Provides a CUDA backend for the PyTorch deep learning framework 336
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
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432
pytorchbearer/torchbearer A PyTorch model fitting library designed to simplify the process of training deep learning models. 636
inisis/brocolli Converts PyTorch models to various formats for deployment and testing in deep learning frameworks. 341
pistony/torch-toolbox A collection of reusable utility functions and classes to simplify PyTorch development 417
ahmedfgad/torchga Trains PyTorch models using the Genetic Algorithm with PyGAD 96
bloodaxe/pytorch-toolbelt A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. 1,520
eduardoleao052/js-pytorch A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. 1,084
erotemic/netharn A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments. 39
rentruewang/koila A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution 1,821
amdegroot/pytorch-containers A collection of PyTorch implementations of Torch Table Layers to simplify the transition from old Torch architectures. 89
metaopt/torchopt An efficient library for differentiable optimization built on top of PyTorch. 544