Supplementary components to accelerate research and development in PyTorch.
Project description
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.8+
- PyTorch 1.10+
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
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pytorch_pfn_extras-0.7.7.tar.gz
(162.5 kB
view details)
Built Distribution
File details
Details for the file pytorch_pfn_extras-0.7.7.tar.gz.
File metadata
- Download URL: pytorch_pfn_extras-0.7.7.tar.gz
- Upload date:
- Size: 162.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 11164018ca8bb7fccd6cfcc392ce28408f51c54d8fd38ad5d8a7322500d58622 |
|
| MD5 | 3558012d7b564d0409550ffe93cf8328 |
|
| BLAKE2b-256 | d832123d73d7ba0550677676fd4da1f1e63a93413d4e908b4b6c97f09474e373 |
File details
Details for the file pytorch_pfn_extras-0.7.7-py3-none-any.whl.
File metadata
- Download URL: pytorch_pfn_extras-0.7.7-py3-none-any.whl
- Upload date:
- Size: 225.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 6ea32a97b22c879461200f4792bbcf15174138aed41d386a22030c16d9d2a404 |
|
| MD5 | 203b8f3afd12cd8181818a2dc92f4a87 |
|
| BLAKE2b-256 | 54526c361bdbb32419cacf99e9b8135249a5acd821d60c116333a9d3402e1c02 |
