3 ms·

Any decent ML library should have support for GPU nowadays. Or is this just a proof of concept to practice a bit with Rust? I am not that used to Rust yet but

Any decent ML library should have support for GPU nowadays. Or is this just a proof of concept to practice a bit with Rust? I am not that used to Rust yet but I wonder, is it easy to write generic (templated?) code which works both on GPU and on CPU? Is there any GPU (CUDA) based ML lib in Rust? I searched a bit and found only this (and related): https://internals.rust-lang.org/t/single-source-gpu-support/898 https://internals.rust-lang.org/t/single-source-gpu-support/... Or maybe this, although this is only indirect: https://github.com/arrayfire/arrayfire-rust https://github.com/arrayfire/arrayfire-rust Also, next topics for ML libs: multi-GPU, distributed computing, ...

mjhirn 11y ago
There is Leaf[1], the Hacker's Machine Learning Framework, which has GPU (CUDA, OpenCL) support for Machine Learning in Rust. James linked to it in the community section of his post, at the end. [1]: https://github.com/autumnai/leaf https://github.com/autumnai/leaf
- dharma1 11y ago
  Was looking at leaf yesterday, looks good. How good is openCL perf with Leaf compared to CUDA?
  - mjhirn 11y ago
    We would love to test that and release the performance on Deep Learning Benchmarks[1] or convnet-benchmarks[2]. But so far no convolution/NN-related OpenCl kernels are linked[3], although it might be quite easy, if the kernels exist. [1]: https://github.com/autumnai/deep-learning-benchmarks https://github.com/autumnai/deep-learning-benchmarks [2]: https://github.com/soumith/convnet-benchmarks https://github.com/soumith/convnet-benchmarks [3]: https://github.com/autumnai/collenchyma-nn https://github.com/autumnai/collenchyma-nn
kobyszcze 11y ago
You could do a lot of ML without GPU support. Especially if you are dealing with sparse data (like text).