pytorch / ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
☆4,586Updated this week
Alternatives and similar repositories for ignite:
Users that are interested in ignite are comparing it to the libraries listed below
- A scikit-learn compatible neural network library that wraps PyTorch☆5,964Updated 2 weeks ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,913Updated last month
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,027Updated 11 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,537Updated 2 weeks ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,081Updated 10 months ago
- Model interpretability and understanding for PyTorch☆5,080Updated this week
- A small package to create visualizations of PyTorch execution graphs☆3,299Updated last month
- Train AI models efficiently on medical images using any framework☆1,870Updated 8 months ago
- A lightweight library for PyTorch training tools and utilities☆1,685Updated this week
- Accelerated deep learning R&D☆3,325Updated 10 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,741Updated last week
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,714Updated 2 weeks ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,810Updated last year
- Debugging, monitoring and visualization for Python Machine Learning and Data Science☆3,433Updated last year
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,088Updated 2 months ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,542Updated 3 years ago
- Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.☆4,281Updated 2 months ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,910Updated last year
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,278Updated this week
- cuML - RAPIDS Machine Learning Library☆4,406Updated this week
- A curated list of awesome self-supervised methods☆6,221Updated 7 months ago
- Models, data loaders and abstractions for language processing, powered by PyTorch☆3,526Updated this week
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.☆29,003Updated this week
- Serve, optimize and scale PyTorch models in production☆4,294Updated this week
- Make huge neural nets fit in memory☆2,758Updated 4 years ago
- starter from "How to Train a GAN?" at NIPS2016☆11,516Updated 3 years ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,389Updated 2 weeks ago
- PyTorch extensions for high performance and large scale training.☆3,260Updated last month
- A Python toolbox for performing gradient-free optimization☆3,999Updated this week
- View model summaries in PyTorch!☆2,693Updated last week