Hi Benjamin,
I do not think you can set per job memory configuration in a shared cluster.
The reason is that if different jobs share the same TM, there are resources that are shared between them, e.g, network buffers.
If you are ok with having a separate cluster per job then this will allow you to set per job memory quotas, because this would make
the total TM memory quota specification correspond to the per job ones.
In [1] you can find the available per job options and on [2] the cluster-wide ones.