#tensorflow #metal #neural-network #kernel

tract-metal

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

3 releases

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

#1088 in Machine learning

Download history 7/week @ 2024-10-08 1/week @ 2024-10-15 5/week @ 2024-10-22 1/week @ 2024-11-12 6/week @ 2024-11-19 1/week @ 2024-11-26 142/week @ 2024-12-03 37/week @ 2024-12-10 3/week @ 2024-12-17 185/week @ 2025-01-07 77/week @ 2025-01-14 40/week @ 2025-01-21

302 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

~17–25MB
~348K SLoC