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