Supplementary components to accelerate research and development in PyTorch.

pip install pytorch-pfn-extras

# Use `[onnx]` to use onnx submodule like:
#  pip install "pytorch-pfn-extras[onnx]"

### Optinal dependencies
# For PlotReport / VariableStatisticsPlot extensions
pip install matplotlib

# For IgniteExtensionsManager
pip install pytorch-ignite torchvision

# For CuPy interoperability (see: https://docs.cupy.dev/en/stable/install.html)
pip install cupy  # or cupy-cudaXXX

• Python 3.5+
• PyTorch 1.6+

Optional dependencies:

• CuPy 8.0+ for PyTorch/CuPy interoperatbility

• Extensions Manager
• Reporting
• Lazy Modules
• Distributed Snapshot
• Config System
• ONNX Utils
• CUDA Utils (CuPy Interoperability)

• Custom training loop
• Ignite integration

You can contribute to this project by sending a pull request. After approval, the pull request will be merged by the reviewer.

Before making a contribution, please confirm that:

• Code quality stays consistent across the script, module or package.
• Code is covered by unit tests.
• API is maintainable.

MIT License