/cupytorch

A small framework mimics PyTorch using CuPy or NumPy

Primary LanguagePythonMIT LicenseMIT

Watchers