Feat: Add Softmax forward and backward passes

Rust-GPU / Rust-CUDA

Ecosystem of libraries and tools for writing and executing fast GPU code fully in Rust.

Apache License 2.0

3.16k stars 120 forks source link

Closed frjnn closed 2 years ago