FlexBox в котором компоненты могут растягиваться по ширине

Нужно чтобы компоненты шли друг за другом, при этом они оставались квадратными, то есть, есть у меня 6 кнопок все они квадратные, в вертикальном положении экрана может помещается 2 кнопки а следующие перенесутся на другую строку, но если перевернуть экран в горизонтальное положение то в первой строке поместиться уже 3 кнопки из-за того что места хватает для всех 3х кнопок. Похожую штуки я вижу в Material components. Но как её реализовать я не знаю. Я думал что можно просто вставить FlexBox и задать компонентам вес но так не работает.