pub fn pbc_distance_gpu(a: [f32; 3], b: [f32; 3], box_size: [f32; 3]) -> f32
Minimum-image distance between two atoms under periodic boundary conditions.
Applies the minimum image convention along each box axis.