Machine Learning in Rust

Re:Machine Learning in Rust by albertzeyer on Tuesday March 08, 2016 @07:22AM (#11244808)

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/...

Or maybe this, although this is only indirect: https://github.com/arrayfire/arrayfire-rust

Also, next topics for ML libs: multi-GPU, distributed computing, ...

[ Parent | on HN ]

Re:Machine Learning in Rust by mjhirn on Tuesday March 08, 2016 @07:35AM (#11244847)

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

[ Parent | on HN ]

Re:Machine Learning in Rust by dharma1 on Tuesday March 08, 2016 @07:54AM (#11244905)

Was looking at leaf yesterday, looks good. How good is openCL perf with Leaf compared to CUDA?

[ Parent | on HN ]

Re:Machine Learning in Rust by mjhirn on Tuesday March 08, 2016 @08:21AM (#11245009)

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

[2]: https://github.com/soumith/convnet-benchmarks

[3]: https://github.com/autumnai/collenchyma-nn

[ Parent | on HN ]

Re:Machine Learning in Rust by kobyszcze on Tuesday March 08, 2016 @11:47AM (#11246424)

You could do a lot of ML without GPU support. Especially if you are dealing with sparse data (like text).

[ Parent | on HN ]