Swift for TensorFlow

我们诚邀您加入 Swift for TensorFlow 开发社区!

Swift for TensorFlow 是一种新的机器学习模型开发方式,它将 TensorFlow 的强大功能直接集成到 Swift 编程语言中。借助 Swift,您可以编写以下命令式代码,Swift 会自动将其转换为单个 TensorFlow 图,并在 CPU、GPU 和 TPU 上运行它(可以实现 TensorFlow 会话的全部性能)。

import TensorFlow

var x = Tensor<Float>([[1, 2], [3, 4]])

for i in 1...5 {
  x += x ⊗ x
}

print(x)

Swift 兼具 Eager Execution 的灵活性和图与会话的卓越性能。Swift 会在后台为您分析 Tensor 代码并自动构建图。Swift 还会在运行您的代码之前检测类型错误和形状不匹配问题,并且内置了自动微分功能。我们认为,机器学习工具非常重要,理应配备一流的语言和编译器

开放源代码

我们已将 Swift for TensorFlow 作为开源项目发布在 GitHub 上!

我们的文档库包含项目概述,以及对特定领域进行深入解释的技术论文。此外还有关于如何安装预构建软件包的说明(适用于 macOS 和 Ubuntu)以及简单的使用教程

今后,我们将采用开放式设计模式,并且所有讨论都将会公开。

在此注册即可加入社区的 Google 网上论坛,我们将在该论坛中发布公告并进行一般性讨论。