rename n_particles to batch_size