ignite
Neural network trainer
A high-level library to help with training and evaluating neural networks in PyTorch
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
5k stars
59 watching
620 forks
Language: Python
last commit: 2 months ago
Linked from 3 awesome lists
closemberdeep-learninghacktoberfestmachine-learningmetricsneural-networkpythonpytorch
academic/awesome-datascience
bharathgs/awesome-pytorch-list
krzjoa/awesome-python-data-scienceRelated projects:
| Repository | Description | Stars |
|---|---|---|
| pytorch/glow | A compiler and execution engine for neural networks that generates optimized code for hardware accelerators | 3,247 |
| pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,978 |
| pytorch/examples | A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. | 22,530 |
 nvidia/apex | Tools for streamlined mixed precision and distributed training in PyTorch | 8,460 |
| pytorch/torchtitan | A native PyTorch library for training large language models using distributed parallelism and optimization techniques. | 2,765 |
 victoresque/pytorch-template | An open-source PyTorch deep learning project template providing a structured environment for building and training various models. | 4,784 |
 yunjey/pytorch-tutorial | A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. | 30,401 |
 szagoruyko/pytorchviz | A tool to visualize PyTorch execution graphs and traces | 3,251 |
| pytorch/tnt | A collection of tools and utilities for building and training neural networks with PyTorch. | 1,669 |
 facebookresearch/fairscale | A PyTorch extension library that provides high-performance and large-scale training techniques. | 3,210 |
 ahkarami/deep-learning-in-production | A collection of notes and references on deploying deep learning models in production environments | 4,313 |
 dmmiller612/sparktorch | A PyTorch implementation on Apache Spark for distributed deep learning model training and inference. | 339 |
| facebookresearch/dinov2 | A PyTorch implementation of a self-supervised learning method for learning robust visual features without supervision. | 9,425 |
 pyg-team/pytorch_geometric | A PyTorch-based library for training and applying Graph Neural Networks to structured data | 21,597 |
 lightning-ai/pytorch-lightning | A framework for building and training AI models on multiple GPUs or TPUs with minimal code changes. | 28,636 |