apex
by NVIDIA
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
AI summary
Training toolkit
Tools for streamlined mixed precision and distributed training in PyTorch
- stars
- 8.5K
- forks
- 1.4K
- watching
- 100
- awesome lists
- 2
Similar projects
Found by comparing what the projects do, not just their names.
Model trainer
A tool to simplify training and deployment of PyTorch models on various devices and configurations
Training framework
A PyTorch extension library that provides high-performance and large-scale training techniques.
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.
pytorch/ignite4.6K
Neural network trainer
A high-level library to help with training and evaluating neural networks in PyTorch
PyTorch examples
A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning.
Optical flow estimator
A deep learning framework implementing an optical flow estimation model using PyTorch.
Audio toolkit
A PyTorch module providing tools and functions for audio signal processing
Neural Network Training Demo
An example project demonstrating how to train and deploy a neural network in Python and C++ using PyTorch 1.0
Graph visualizer
A tool to visualize PyTorch execution graphs and traces
kwea123/nerf_pl2.7K
NeRF library
An implementation of NeRF in PyTorch-Lightning for 3D reconstruction and view synthesis
bytedance/byteps3.6K
Distributed DL framework
A high-performance distributed deep learning framework supporting multiple frameworks and networks
PyTorch Models
Provides pre-trained PyTorch models for various computer vision tasks and datasets.
Pipeline Parallelism Library
A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients.
Speech Synthesis Toolkit
This PyTorch implementation provides a toolkit for speech synthesis using a deep neural network architecture.
Featured in 2 awesome lists
Each link jumps to the spot where the list mentions apex.
