pub trait Softmax { type Output; // Required methods fn softmax(&self) -> Self::Output; fn softmax_derivative(&self) -> Self::Output; }