pub(crate) fn py_infonce_loss( anchors: PyReadonlyArray1<'_, f64>, positives: PyReadonlyArray1<'_, f64>, n: usize, d: usize, temperature: f64, ) -> f64