accelerate
by huggingface
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
AI summary
Model trainer
A tool to simplify training and deployment of PyTorch models on various devices and configurations
- stars
- 8.1K
- forks
- 989
- watching
- 97
- awesome lists
- 3
Similar projects
Found by comparing what the projects do, not just their names.
nvidia/apex8.5K
Training toolkit
Tools for streamlined mixed precision and distributed training in PyTorch
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
LLM trainer
A native PyTorch library for training large language models using distributed parallelism and optimization techniques.
Model optimizer
A toolkit for optimizing and accelerating the training and inference of machine learning models on various hardware platforms.
pytorch/serve4.2K
Model Server
A tool for serving and scaling PyTorch models in production environments
Training framework
A PyTorch extension library that provides high-performance and large-scale training techniques.
ODE solver library
A PyTorch implementation of differentiable ODE solvers with GPU support and efficient backpropagation
Neural network trainer
A high-level library to help with training and evaluating neural networks in PyTorch
Face parser
Implementation of a face parsing model using PyTorch and a modified BiSeNet architecture.
Object Detector
An implementation of a deep learning-based object detection system in PyTorch.
DL framework
A PyTorch framework for accelerating deep learning research and development by focusing on reproducibility, rapid experimentation, and codebase reuse.
SE-Nets
An implementation of Squeeze-and-Excitation Networks (SE-Nets) for deep learning image classification tasks
Parameter adaptation
An efficient method for fine-tuning large pre-trained models by adapting only a small fraction of their parameters
Object detector
An object detection implementation built on top of PyTorch, supporting multi-image batch training and multiple GPUs.
Neural network compiler
A compiler and execution engine for neural networks that generates optimized code for hardware accelerators
Featured in 3 awesome lists
Each link jumps to the spot where the list mentions accelerate.
