flashinfer.cake\_sampling.choose\_stage1
========================================

.. currentmodule:: flashinfer.cake_sampling

.. autofunction:: choose_stage1