pytorch-pfn-extras

pytorch-pfn-extras (PPE) is a collection of supplementary components to accelerate research and development in PyTorch.

• User Guide
  • Quick Start
  • Trainer
  • Extensions
  • Utilities
• API Reference
  • Package
  • Training Loop
  • Distributed Training
  • Check Pointing
  • Lazy Modules
  • ONNX
  • Datasets
  • Config
  • NumPy/CuPy Compatibility