Swift for TensorFlow APIs

This repository hosts Swift for TensorFlow's deep learning library, available both as a part of the Swift for TensorFlow toolchain and as a Swift package.

Requirements

A latest Swift for TensorFlow toolchain.

Usage

A Swift for TensorFlow toolchain is required to use this package. Add the following to your Swift package manifest.

packages: [
    .package(url: "https://github.com/tensorflow/swift-apis.git")
]

To get started, simply import TensorFlow in your Swift code.

import TensorFlow

struct Model: Layer {
    var l1, l2: Dense<Float>

    init(hiddenSize: Int) {
        l1 = Dense<Float>(inputSize: 2, outputSize: hiddenSize, activation: relu)
        l2 = Dense<Float>(inputSize: hiddenSize, outputSize: 1, activation: relu)
    }

    @differentiable(wrt: (self, input))
    func applied(to input: Tensor<Float>) -> Tensor<Float> {
        let h1 = l1.applied(to: input)
        return l2.applied(to: h1)
    }
}

let optimizer = SGD<Classifier, Float>(learningRate: 0.02)
var classifier = Model(hiddenSize: 4)
let x: Tensor<Float> = ...
let y: Tensor<Float> = ...

for _ in 0..<1000 {
    let 𝛁model = classifier.gradient { classifier -> Tensor<Float> in
        let ŷ = classifier.applied(to: x)
        let loss = meanSquaredError(predicted: ŷ, expected: y)
        print("Loss: \(loss)")
        return loss
    }
    optimizer.update(&classifier.allDifferentiableVariables, along: 𝛁model)
}

Building

swift build

Bugs

Please report bugs and feature requests using GitHub issues in this repository.

Community

Discussion about Swift for TensorFlow happens on the swift@tensorflow.org mailing list.

Contributing

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.

Code of Conduct

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.

jekbradbury / swift-apis