5 ms·
This is cool (and congrats on writing your first Rust lib!), but Metal/Cuda support is a must for serious local usage.
by dvt 2y ago
This is cool (and congrats on writing your first Rust lib!), but Metal/Cuda support is a must for serious local usage.
- littlestymaar 2y agoUsing Cuda is a non starter because it would go against the purpose of this project, but I (not the main author but contributor) am experimenting with wgpu to get some kind of GPU acceleration. I'm not sure it goes anywhere though, because the main author want to keep the complexity under control.
- tormeh 2y agowgpu would be awesome. Too little ML software out there is hardware-agnostic.
- littlestymaar 2y agoThat's exactly my feeling and that's why I started working on it.