pub fn validate_momentum(momentum: f32) -> PyResult<()>
Validate that momentum is in valid range [0, 1]