TODO: 2 schedulers, 4 workers and batch sampling benchmark