PyPI page
Home page
Author:
Locomp Contributors
License:
Apache-2.0
Summary:
A Python GPU kernel compiler for Apple Silicon (Metal), NVIDIA (CUDA), AMD (ROCm/HIP), and RISC-V — write kernels once in Python, run on all four backends
Latest version:
1.0.0
Required dependencies:
numpy
|
pyobjc-framework-metal
|
pyobjc-framework-metalperformanceshaders
Optional dependencies:
pytest
|
pytest-benchmark
|
ruff
Downloads last day:
0
Downloads last week:
34
Downloads last month:
231