Setup development environments is always pain in the ass, happy to see a solution for model training