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.02k stars 115 forks source link

Closed frjnn closed 2 years ago