2 releases
| 0.21.8 | Dec 5, 2024 |
|---|---|
| 0.21.7 | Sep 23, 2024 |
#1062 in Machine learning
186 downloads per month
Used in tract
1.5MB
37K
SLoC
tract-metal
Updating Metal Flash Attention library
git clone https://github.com/philipturner/metal-flash-attention.git
cd metal-flash-attention
# for iOS
swift build.swift --platform iOS --xcode-path /Applications/Xcode.app
cp build/lib/libMetalFlashAttention.metallib path/to/tract/metal/src/kernels/libMetalFlashAttention-ios.metallib
# for MacOS
swift build.swift --platform macOS --xcode-path /Applications/Xcode.app
cp build/lib/libMetalFlashAttention.metallib path/to/tract/metal/src/kernels/libMetalFlashAttention-macos.metallib
Dependencies
~17–25MB
~347K SLoC