Then you'll be happy to learn about Swift for Tensorflow:

https://www.tensorflow.org/swift/

And no.. it's not simply a Swift wrapper for tensorflow, it's literally Tensorflow (or generally speaking: differentiable calculation graphs) built into the Swift language.