pub fn register(registry: &mut KernelRegistry)
Register softmax shader sources with the given kernel registry.