#tensor #cuda #machine-learning

ug-metal

Micro compiler for tensor operations

2 unstable releases

0.1.0 Jan 16, 2025
0.0.2 Oct 27, 2024

#1597 in Machine learning

Download history 306/week @ 2024-11-30 400/week @ 2024-12-07 329/week @ 2024-12-14 204/week @ 2024-12-21 323/week @ 2024-12-28 375/week @ 2025-01-04 612/week @ 2025-01-11 501/week @ 2025-01-18 511/week @ 2025-01-25 457/week @ 2025-02-01 662/week @ 2025-02-08 743/week @ 2025-02-15 676/week @ 2025-02-22 623/week @ 2025-03-01 627/week @ 2025-03-08 781/week @ 2025-03-15

2,840 downloads per month
Used in 34 crates (4 directly)

MIT/Apache

315KB
8K SLoC

Rust 7K SLoC // 0.0% comments Metal Shading Language 1K SLoC // 0.1% comments

ug

rust ci badge

Experimental compiler for deep-learning models inspired by triton, tinygrad, and micrograd.

Dependencies

~11–17MB
~191K SLoC