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.9+
- PyTorch 1.13+
Optional dependencies:
- CuPy 8.0+ for PyTorch/CuPy interoperatbility
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)
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