PyPI page
Home page
Author:
Christopher Landschoot
Summary:
Fast drop-in scaled_dot_product_attention for Apple silicon (MPS) — wraps Apple's native MPSGraph SDPA with a zero-copy C++ bridge. 5-7x inference, 2-2.5x training.
Latest version:
0.2.0
Required dependencies:
ninja
|
numpy
|
psutil
|
pyobjc-core
|
pyobjc-framework-metal
|
pyobjc-framework-metalperformanceshadersgraph
|
torch
Optional dependencies:
build
|
matplotlib
|
pytest
|
scipy
|
twine
Downloads last day:
12
Downloads last week:
51
Downloads last month:
342