torchgpipe
by kakaobrain
A GPipe implementation in PyTorch
AI summary
Pipeline Parallelism Library
A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients.
- stars
- 820
- forks
- 100
- watching
- 33
Similar projects
Found by comparing what the projects do, not just their names.
GA trainer
Trains PyTorch models using a genetic algorithm
Neural net trainer
A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities
Optimization library
An efficient library for differentiable optimization built on top of PyTorch.
PyTorch C++ implementation
A C++ implementation of PyTorch tutorials
Batching library
A PyTorch module for dynamic batching and optimized computation on deep neural networks
Deep learning library
A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers.
Image generator
Implementation of a deep learning model for generating high-quality images with improved stability and variation.
Model trainer
A PyTorch model fitting library designed to simplify the process of training deep learning models.
Metrics library
A collection of common machine learning evaluation metrics implemented in PyTorch
ML toolkit
A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development.
Deep learning library
Provides a CUDA backend for the PyTorch deep learning framework
Image synthesis library
An implementation of cascaded refinement networks for generating photorealistic images from semantic layouts
Deep learning trainer
Provides a flexible and configurable framework for training deep learning models with PyTorch.
Reinforcement Learning Environment
Enables training of reinforcement learning models with PyTorch on Godot Engine using shared memory
Tensor library
Bindings for PyTorch's tensor library in OCaml for GPU acceleration and automatic differentiation
