A gpu accelerated neural network Rust crate.
MIT License
Implementation of the RWKV language model in pure WebGPU/Rust.
Deep learning in Rust, with shape checked tensors and neural networks
Machine learning crate for Rust
Efficient inference ML framework written in rust