pytorch/helion
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
Stars: 772Language: Python
Give AlbumentationsX a star on GitHub — it powers this leaderboard
Star on GitHubA Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.