TensorFlow for Java

在 TensorFlow.org 上查看 查看 GitHub 仓库

TensorFlow Java 可以在任何 JVM 上运行,用于构建、训练和运行机器学习模型。它提供了一系列实用程序和框架,可以帮助您完成数据科学家和开发人员在该领域中常见的任务。Java 和其他 JVM 语言(如 Scala 或 Kotlin)在全球范围内的小型到大型企业中被广泛使用,这使得 TensorFlow 成为大规模采用机器学习的战略选择。

仓库

在早期,TensorFlow 的 Java 语言绑定托管在 主 TensorFlow 仓库 中,并且仅在核心库的新版本准备发布时才会发布,这种情况每年只发生几次。现在,所有与 Java 相关的代码都已移至此仓库,以便它可以独立于官方 TensorFlow 版本进行演进和发布。此外,大多数构建任务已从 Bazel 迁移到 Maven,这对于大多数 Java 开发人员来说更为熟悉。

以下描述了仓库的布局及其不同的工件

  • tensorflow-core

    • 所有构建 TensorFlow for Java 核心语言绑定的工件
    • 目标受众:在 TensorFlow 之上提供自己的 API 或框架的项目,只需要一个薄层来从 JVM 访问 TensorFlow 运行时
  • tensorflow-framework

    • 使用 TensorFlow 构建和训练神经网络的主要 API
    • 目标受众:神经网络开发人员
  • ndarray

    • 用于 n 维数据 I/O 操作的通用实用程序库
    • 由 TensorFlow 使用,但不依赖于 TensorFlow
    • 目标受众:任何需要 Java n 维数组实现的开发人员,无论他们是否将其与 TensorFlow 一起使用

沟通

此仓库由 TensorFlow JVM 特别兴趣小组 (SIG) 维护。您可以通过订阅 [email protected] 邮件列表轻松加入该小组,或者您也可以直接向此仓库发送拉取请求和提出问题。还有一个 sig-jvm Gitter 频道