torchfold
Batching library
A PyTorch module for dynamic batching and optimized computation on deep neural networks
Tools for PyTorch
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 |