dstack is now supported in Transformer Lab. More details here.