pytorch-pfn-extras

Supplementary components to accelerate research and development in PyTorch.

Installation

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

Requirements

• Python 3.9+
• PyTorch 1.13+

Optional dependencies:

• CuPy 8.0+ for PyTorch/CuPy interoperatbility

Documentation

Refer to Read The Docs for the complete documentation.

Below are some quick-links to the most important features of the library.

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

Examples

• Custom training loop
• Ignite integration

Contribution Guide

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.

License

MIT License