Hello Laurent,
There is no way to do that with the heat_tree_matrix function, but its a good idea for a feature in the future. I can imagine two ways of doing it now:
1) make each plot individually and combine them into a single plot (this is how heat_tree_matrix works). In the example analysis you linked, each plot would be made in the same way as the plot in the "Comparing two treatments/groups" section, except you would not add labels or legends and always use the same random seed (to make the layout always the same). The labelled key plot would be the same tree but with the node color set to grey. There are multiple pacakges that let you combine plots, such as cowplot.
2) Kindof a hack, but works. Edit the output of a full heat_tree_matrix by hand using a program like inkscape (free) or photoshop. If you save the plot as SVG or PDF, then you can rearrange and resize parts without loss of quality. You can just delete the plots you dont want and scale the others to look nice.
Let me know if you have questions.
Best,
Zach