Gumbel-softmax对于硬件的要求是怎样的呢？和用大模型来选择相比优势在哪里？