Function fits_in_memory 

pub fn fits_in_memory(
    count: usize,
    dim: usize,
    memory_limit_bytes: usize,
) -> bool

Checks whether count f32 embeddings of length dim fit under the memory limit.