#cuda #tensor #machine-learning

ug-metal

Micro compiler for tensor operations

3 releases (breaking)

new 0.2.0 Apr 2, 2025
0.1.0 Jan 16, 2025
0.0.2 Oct 27, 2024

#52 in #cuda

Download history 368/week @ 2024-12-11 277/week @ 2024-12-18 272/week @ 2024-12-25 324/week @ 2025-01-01 511/week @ 2025-01-08 589/week @ 2025-01-15 430/week @ 2025-01-22 556/week @ 2025-01-29 511/week @ 2025-02-05 700/week @ 2025-02-12 769/week @ 2025-02-19 623/week @ 2025-02-26 591/week @ 2025-03-05 774/week @ 2025-03-12 771/week @ 2025-03-19 677/week @ 2025-03-26

2,908 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

~11–17MB
~190K SLoC