Function write_batch 

pub fn write_batch(
    epochs: &[Array2<f32>],
    positions: &[Array2<f32>],
    path: &Path,
) -> Result<()>

Write preprocessed epochs to batch.safetensors.

epochs[e]: [C, 1280] positions[e]: [C, 3]