That uses a 2mm driver.
I use a 1.5mm driver for nearly all my atties, personally.
To achieve that, 3mm posts get grub screws, 2.5mm posts get button head screws and 2mm posts get cap head screws.
So I'd get button heads for that, personally.
McMaster-Carr
Not 316, but IME, they'll be just...