Thanks for posting in!
You can achieve what you have in mind by using Essential Grid or The Grid plugin. You can create a grid that display items of the same category in a masonry layout. You can check out some of the grid examples from here:
You can install this plugins by going to X > Validation > Extensions. And to know more about this plugins, please check this out: https://theme.co/apex/forum/t/extension-essential-grid/68, https://theme.co/apex/forum/t/extension-the-grid/74
Hope this helps.