stoke
Lightweight wrapper for PyTorch that provides a simple unified interface for context switching between devices (CPU, GPU), distributed modes (DDP, Horovod), mixed-precision (AMP, Apex), and extension…
Installation
In a virtualenv (see these instructions if you need to create one):
pip3 install stoke
Dependencies
Releases
| Version | Released | Bullseye Python 3.9 |
Bookworm Python 3.11 |
Trixie Python 3.13 |
Files |
|---|---|---|---|---|---|
| 0.2.1 | 2022-05-20 | ||||
|
|
|||||
| 0.2.0 | 2021-10-26 | ||||
|
|
|||||
| 0.1.2 | 2021-08-02 | ||||
|
|
|||||
| 0.1.1 | 2021-07-28 | ||||
|
|
|||||
| 0.1.0 | 2021-06-23 | ||||
|
|||||
Issues with this package?
- Search issues for this package
- Package or version missing? Open a new issue
- Something else? Open a new issue