Comment by willglynn

You might be interested in https://burn.dev, a Rust machine learning framework. It has CUDA and ROCm backends among others.