1 2 3 4 5 6 7 8 9 10
//! Machine learning kernels for GPU pub mod activation; pub mod pooling; pub mod softmax; // Re-export commonly used activation functions for easy access pub use activation::{ GeluKernel, LeakyReluKernel, ReluKernel, SigmoidKernel, SwishKernel, TanhKernel, };