#59 · Primary category: Deep Learning Frameworks
ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Project last updated:08/28/26
GitHub Stars
4.8K
Forks
719
Contributors
263
License
BSD-3-Clause
Why we included this project
Writing PyTorch training scripts often means repeating the same setup: the loop, checkpointing, metric tracking, and early stopping. Ignite replaces that with a small engine that runs your training and evaluation steps and lets you attach handlers to specific events like iteration or epoch end, so the logic remains explicit and easy to follow. It ships with built-in metrics for accuracy, precision, and confusion matrices, which saves you from rolling your own and makes results comparable across runs. For research code you'll want to revisit or production pipelines where control matters, this sits comfortably between raw PyTorch and heavier frameworks.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.