numr 0.5.1

High-performance numerical computing with multi-backend GPU acceleration (CPU/CUDA/WebGPU)
Documentation 
1
2
3
4
5
6
7
8
9
10
11

//! Activation CUDA kernel launchers
//!
//! Split into submodules:
//! - `elementwise` - relu, sigmoid, silu, gelu, leaky_relu, elu
//! - `softmax` - softmax forward + backward (last-dim and non-last-dim)

mod elementwise;
mod softmax;

pub use elementwise::*;
pub use softmax::*;