Just be aware flexbox might cause issues with Internet Explorer. There are two other options you could use. Both are listed in this blog post.
The easier being Sergie’s column solution. The more complex being Masonry.
1 Like