| Metadata-Version: |
2.1 |
| Name: |
stoke |
| Version: |
0.2.1 |
| Summary: |
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 extensions (fairscale, deepspeed). |
| Author: |
FMR LLC |
| Home-Page: |
https://github.com/fidelity/stoke |
| Project-Url: |
Source, https://github.com/fidelity/stoke |
| Project-Url: |
Documentation, https://fidelity.github.io/stoke/ |
| Project-Url: |
Bug Tracker, https://github.com/fidelity/stoke/issues |
| Keywords: |
deep learning,pytorch,AI,gpu,ddp,horovod,deepspeed,fairscale,distributed,fp16,mixed-precision,apex,amp |
| Classifier: |
Development Status :: 4 - Beta |
| Classifier: |
Environment :: GPU :: NVIDIA CUDA |
| Classifier: |
Intended Audience :: Developers |
| Classifier: |
Natural Language :: English |
| Classifier: |
License :: OSI Approved :: Apache Software License |
| Classifier: |
Programming Language :: Python :: 3 |
| Classifier: |
Programming Language :: Python :: 3 |
| Classifier: |
Programming Language :: Python :: 3.6 |
| Classifier: |
Programming Language :: Python :: 3.7 |
| Classifier: |
Programming Language :: Python :: 3.8 |
| Classifier: |
Programming Language :: Python :: 3.9 |
| Classifier: |
Operating System :: OS Independent |
| Classifier: |
Topic :: Scientific/Engineering |
| Classifier: |
Topic :: Scientific/Engineering :: Artificial Intelligence |
| Classifier: |
Topic :: Software Development :: Libraries :: Python Modules |
| Requires-Python: |
>=3.6 |
| Requires-Dist: |
attrs (>=20.3.0) |
| Requires-Dist: |
deepspeed (>=0.6.4) |
| Requires-Dist: |
fairscale (>=0.4.6) |
| Requires-Dist: |
horovod (>=0.21.2) |
| Requires-Dist: |
torch (>=1.8.1) |
| Requires-Dist: |
mypy-extensions; python_version < "3.8" |
| Requires-Dist: |
mpi4py (==3.0.3); extra == "mpi" |
| Provides-Extra: |
mpi |
| Description-Content-Type: |
text/markdown |