#cuda #tensor #machine-learning #ug

ug-metal

Micro compiler for tensor operations

6 releases (breaking)

new 0.4.0 Apr 15, 2025
0.3.1 Apr 12, 2025
0.2.0 Apr 2, 2025
0.1.0 Jan 16, 2025
0.0.2 Oct 27, 2024

#59 in #cuda

Download history 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 623/week @ 2025-03-08 857/week @ 2025-03-15 685/week @ 2025-03-22 698/week @ 2025-03-29 636/week @ 2025-04-05 711/week @ 2025-04-12

2,820 downloads per month
Used in 35 crates (5 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

~10–16MB
~182K SLoC