JSFiddle - React, Tailwind, and code Playground

HTML

<div class="box">
  <div>One</div>
  <div>Two</div>
  <div>Three</div>
  <div>Four</div>
  <div>Five</div>
  <div>Six</div>
</div>

CSS

.box {
  display: grid;
  grid-template-columns: repeat(6, 50px);
  grid-auto-rows: 50px;
  grid-gap: 5px;
}

@media ( max-width: 800px) {
  .box {
    grid-template-columns: repeat(3, 50px);
  }
}

@media ( max-width: 600px) {
  .box {
    grid-template-columns: 50px 50px;
  }
}

@media ( max-width: 400px) {
  .box {
    grid-template-columns: 50px;
  }
}

.box div {
  background-color: yellow;
  border: 1px solid gray;
}

* {
  box-sizing: border-box;
}