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.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
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
Built Distribution
File details
Details for the file pytorch_pfn_extras-0.8.1.tar.gz.
File metadata
- Download URL: pytorch_pfn_extras-0.8.1.tar.gz
- Upload date:
- Size: 162.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | c5ec411752a71269176836cce49d12108dc5d861f27c0fbf2b4cf3fdb15534b7 |
|
| MD5 | f877e49fa2538b93221f9bafc3e93e30 |
|
| BLAKE2b-256 | 8f51710af807acca981648f775ed816e9db489c6b68c5602d6a7c7025a39cd5b |
Provenance
The following attestation bundles were made for pytorch_pfn_extras-0.8.1.tar.gz:
Publisher:
publish.yml on pfnet/pytorch-pfn-extras
-
Statement type:
https://in-toto.io/Statement/v1- Predicate type:
https://docs.pypi.org/attestations/publish/v1 - Subject name:
pytorch_pfn_extras-0.8.1.tar.gz - Subject digest:
c5ec411752a71269176836cce49d12108dc5d861f27c0fbf2b4cf3fdb15534b7 - Sigstore transparency entry: 150135540
- Sigstore integration time:
- Predicate type:
File details
Details for the file pytorch_pfn_extras-0.8.1-py3-none-any.whl.
File metadata
- Download URL: pytorch_pfn_extras-0.8.1-py3-none-any.whl
- Upload date:
- Size: 225.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 4f67bcf067b4aba41e50f544ae367bc9f7a712a67b94215e8fe0083861bd2751 |
|
| MD5 | b9948245a463580703a5fa30c912f26b |
|
| BLAKE2b-256 | 0ae9a553c2257f2aa0d9b7e22bb555bd1aa4aa982c633c3392746b41b6cb3134 |
Provenance
The following attestation bundles were made for pytorch_pfn_extras-0.8.1-py3-none-any.whl:
Publisher:
publish.yml on pfnet/pytorch-pfn-extras
-
Statement type:
https://in-toto.io/Statement/v1- Predicate type:
https://docs.pypi.org/attestations/publish/v1 - Subject name:
pytorch_pfn_extras-0.8.1-py3-none-any.whl - Subject digest:
4f67bcf067b4aba41e50f544ae367bc9f7a712a67b94215e8fe0083861bd2751 - Sigstore transparency entry: 150135541
- Sigstore integration time:
- Predicate type:
