#tensorflow #metal #neural-network #kernel

tract-metal

Tiny, no-nonsense, self contained, TensorFlow and ONNX inference

3 releases

new 0.21.9 Jan 8, 2025
0.21.8 Dec 5, 2024
0.21.7 Sep 23, 2024

#1058 in Machine learning

Download history 158/week @ 2024-09-20 37/week @ 2024-09-27 11/week @ 2024-10-04 3/week @ 2024-10-11 4/week @ 2024-10-18 2/week @ 2024-10-25 4/week @ 2024-11-15 3/week @ 2024-11-22 86/week @ 2024-11-29 88/week @ 2024-12-06 9/week @ 2024-12-13 134/week @ 2025-01-03

157 downloads per month
Used in 4 crates (2 directly)

MIT/Apache

1.5MB
38K 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

~16–24MB
~327K SLoC