Function batch_gradient_update 

pub fn batch_gradient_update(
    embeddings: &mut [Array2<f64>],
    gradients: &[Array2<f64>],
    learning_rate: f64,
    l2_reg: f64,
)

Batch gradient update for multiple embedding matrices