Lib.rs

› Science › Machine learning › ug
#tensor #cuda #machine-learning

ug-pyo3

Micro compiler for tensor operations

by Laurent Mazare

1 unstable release

0.0.1 Oct 18, 2024

#175 in #cuda

MIT/Apache

290KB
7.5K SLoC

ug

rust ci badge

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

Dependencies

~36MB
~822K SLoC