We do not have such feature at the moment.
The main limitation I recall for this is that the CUDA driver uses a significant amount of memory that we cannot track. Thus the limit set by the user my not be respected because of this extra memory. This makes the feature much weaker as it has some unexpected behavior for the end user.
If you have a proposal to overcome this issues, please feel free to open a feature request on github !