pytorch/helion

A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

语言：Python · GitHub 星标：977 · Fork：183

在 GitHub 上查看原仓库