I don't think that the main goal of TF on Swift is to train models using Swift. I think it's mainly to deploy them in production on iPhones