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) 进行维护。您只需订阅 jvm@tensorflow.org 网上论坛即可轻松地加入该小组,或者,您也可以直接向该代码库发送拉取请求和提出问题。此外,我们还提供了一个 sig-jvm Gitter 渠道