mempool: fix mlx driver loading