stoke (pypi): 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).
Keywords: deep learning|pytorch|AI|gpu|ddp|horovod|deepspeed|fairscale|distributed|fp16|mixed-precision|apex|amp