@amosella Would you like to post this on https://github.com/pytorch/pytorch/issues 11, since this is not only related to libtorch? We would be able to prioritize general feature requests there based on community needs and available resources.