These packages are available as a package collection, usable in Xcode or SwiftPM.
Accelerated tensor operations and dynamic neural networks based on reverse mode automatic differentiation for every device that can run Swift - from watchOS to Linux
1 package.
Swift Package Index