Take a look at the neural network framework I wrote, which is implemented in Rust + CUDA

The blog post introduces a neural network framework called Lumen, developed using Rust and CUDA. It highlights features such as dynamic autograd, module-style abstraction for models, a flexible precision system, and CPU/CUDA execution paths, ultimately showcasing Lumen's performance and usability for deep learning tasks.