ignite
by pytorch
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
AI summary
Neural network trainer
A high-level library to help with training and evaluating neural networks in PyTorch
- stars
- 4.6K
- forks
- 620
- watching
- 59
- awesome lists
- 3
Similar projects
Found by comparing what the projects do, not just their names.
Neural network compiler
A compiler and execution engine for neural networks that generates optimized code for hardware accelerators
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
PyTorch examples
A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning.
nvidia/apex8.5K
Training toolkit
Tools for streamlined mixed precision and distributed training in PyTorch
LLM trainer
A native PyTorch library for training large language models using distributed parallelism and optimization techniques.
Deep Learning Template
An open-source PyTorch deep learning project template providing a structured environment for building and training various models.
Deep Learning Tutorials
A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage.
Graph visualizer
A tool to visualize PyTorch execution graphs and traces
PyTorch toolkit
A collection of tools and utilities for building and training neural networks with PyTorch.
Training framework
A PyTorch extension library that provides high-performance and large-scale training techniques.
Model deployment notes
A collection of notes and references on deploying deep learning models in production environments
Deep learning framework
A PyTorch implementation on Apache Spark for distributed deep learning model training and inference.
Visual feature learner
A PyTorch implementation of a self-supervised learning method for learning robust visual features without supervision.
Graph neural network library
A PyTorch-based library for training and applying Graph Neural Networks to structured data
AI model builder
A framework for building and training AI models on multiple GPUs or TPUs with minimal code changes.
Featured in 3 awesome lists
Each link jumps to the spot where the list mentions ignite.
