Plugin to add sorting/categories to masonry gallery

I’m looking for a plugin that adds sorting/categories to a masonry gallery.

On this page, for example. I want to add sorting to allow users to only display certain groups of work within the gallery.

I would like to retain the lightbox-on-click functionality.

Any recommendations (free or paid) are much appreciated!