I am trying to create a UI which looks very similar to the one at the top of the Material3 Chip page.

It appears that the best way to do so is to construct a LazyVerticalStaggeredGrid, whose content consists of FilteredChip composables.