pytorch / ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
☆4,523Updated this week
Related projects ⓘ
Alternatives and complementary repositories for ignite
- Model interpretability and understanding for PyTorch☆4,918Updated this week
- A scikit-learn compatible neural network library that wraps PyTorch☆5,873Updated this week
- A small package to create visualizations of PyTorch execution graphs☆3,219Updated 7 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,016Updated 8 months ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,870Updated 2 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,038Updated 7 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,500Updated 3 weeks ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,392Updated last week
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,691Updated this week
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.☆28,337Updated this week
- On the Variance of the Adaptive Learning Rate and Beyond☆2,535Updated 3 years ago
- A lightweight library for PyTorch training tools and utilities☆1,662Updated this week
- Bayesian optimization in PyTorch☆3,096Updated this week
- Train AI models efficiently on medical images using any framework☆1,864Updated 4 months ago
- A hyperparameter optimization framework☆10,852Updated this week
- Automatic architecture search and hyperparameter optimization for PyTorch☆2,371Updated 7 months ago
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,007Updated this week
- Accelerated deep learning R&D☆3,291Updated 7 months ago
- Models, data loaders and abstractions for language processing, powered by PyTorch☆3,513Updated this week
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,796Updated 8 months ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,134Updated this week
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,268Updated last month
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,482Updated this week
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,030Updated 2 years ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,139Updated this week
- Code for visualizing the loss landscape of neural nets☆2,824Updated 2 years ago
- AI conference deadline countdowns☆5,659Updated last month
- Debugging, monitoring and visualization for Python Machine Learning and Data Science☆3,417Updated last year
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,570Updated last week
- PyTorch tutorials and best practices.☆1,656Updated 2 years ago