Function set_memory_limit 

pub fn set_memory_limit(
    active_memory_limit_bytes: usize,
) -> Result<usize, MlxCError>

Applies the graph-evaluation active-memory guidance and returns the previous limit in bytes.