Get a taste of protocol-oriented differentiable programming.
This repository hosts Swift for TensorFlow's deep learning library, available both as a part of Swift for TensorFlow toolchains and as a Swift package.
This library is being automatically integrated in Swift for TensorFlow toolchains. You do not need to add this library as a Swift Package Manager dependency.
Open an empty Colaboratory now to try out Swift, TensorFlow, differentiable programming, and deep learning.
For detailed usage and troubleshooting, see Usage on the Swift for TensorFlow project homepage.
Simply import TensorFlow
to get the full power of TensorFlow.
import TensorFlow
let hiddenSize: Int = 10
struct Model: Layer {
var layer1 = Dense<Float>(inputSize: 4, outputSize: hiddenSize, activation: relu)
var layer2 = Dense<Float>(inputSize: hiddenSize, outputSize: hiddenSize, activation: relu)
var layer3 = Dense<Float>(inputSize: hiddenSize, outputSize: 3, activation: identity)
@differentiable
func callAsFunction(_ input: Tensor<Float>) -> Tensor<Float> {
return input.sequenced(through: layer1, layer2, layer3)
}
}
var classifier = Model()
let optimizer = SGD(for: classifier, learningRate: 0.02)
Context.local.learningPhase = .training
let x: Tensor<Float> = ...
let y: Tensor<Int32> = ...
One way to define a training epoch is to use the Differentiable.gradient(in:)
method.
for _ in 0..<1000 {
let 𝛁model = classifier.gradient { classifier -> Tensor<Float> in
let ŷ = classifier(x)
let loss = softmaxCrossEntropy(logits: ŷ, labels: y)
print("Loss: \(loss)")
return loss
}
optimizer.update(&classifier, along: 𝛁model)
}
Another way is to make use of methods on Differentiable
or Layer
that produce a backpropagation function. This allows you to compose your derivative computation with great flexibility.
for _ in 0..<1000 {
let (ŷ, backprop) = classifier.appliedForBackpropagation(to: x)
let (loss, 𝛁ŷ) = ŷ.valueWithGradient { ŷ in softmaxCrossEntropy(logits: ŷ, labels: y) }
print("Model output: \(ŷ), Loss: \(loss)")
let 𝛁model = backprop(𝛁ŷ)
optimizer.update(&classifier, along: 𝛁model)
}
For more models, go to tensorflow/swift-models.
- Swift for TensorFlow toolchain.
- An environment that can run the Swift for TensorFlow toolchains: Linux 18.04 or macOS with Xcode 10.
$ swift build
$ swift test
Note: CMake support is experimental and under development.
In-tree builds are not supported. The instructions here expect CMake 3.16
or newer, although the minimum required version is 3.15.1. Older releases
will not allow the use of the -B
option to specific the build tree and
require that you are in the location of the build tree (and the -B
option
and its argument are elided).
If swiftc
is not in your PATH
, you must specify the path to it using
-D CMAKE_Swift_COMPILER=
.
cmake -G Ninja -B out -S swift-apis
cmake --build out
Please report bugs and feature requests using GitHub issues in this repository.
Discussion about Swift for TensorFlow happens on the [email protected] mailing list.
We welcome contributions: please read the Contributor Guide to get started. It's always a good idea to discuss your plans on the mailing list before making any major submissions.
In the interest of fostering an open and welcoming environment, we as contributors and maintainers pledge to making participation in our project and our community a harassment-free experience for everyone, regardless of age, body size, disability, ethnicity, gender identity and expression, level of experience, education, socio-economic status, nationality, personal appearance, race, religion, or sexual identity and orientation.
The Swift for TensorFlow community is guided by our Code of Conduct, which we encourage everybody to read before participating.