fidelity / stoke

A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-precision, and PyTorch extensions.
66Updated last year

Alternatives and similar repositories for stoke:

Users that are interested in stoke are comparing it to the libraries listed below