TensorFlow for Java

View on TensorFlow.org View GitHub repository

TensorFlow Java can run on any JVM for building, training and running machine learning models. It comes with a series of utilities and frameworks that help achieve most of the tasks common to data scientists and developers working in this domain. Java and other JVM languages, such as Scala or Kotlin, are frequently used in small-to-large enterprises all over the world, which makes TensorFlow a strategic choice for adopting machine learning at a large scale.

The Repository

In the early days, the Java language bindings for TensorFlow were hosted in the main TensorFlow repository and released only when a new version of the core library was ready to be distributed, which happens only a few times a year. Now, all Java-related code has been moved to this repository so that it can evolve and be released independently from official TensorFlow releases. In addition, most of the build tasks have been migrated from Bazel to Maven, which is more familiar for most Java developers.

The following describes the layout of the repository and its different artifacts:

  • tensorflow-core

    • All artifacts that build up the core language bindings of TensorFlow for Java
    • Intended audience: projects that provide their own APIs or frameworks on top of TensorFlow and just want a thin layer to access the TensorFlow runtime from the JVM
  • tensorflow-framework

    • Primary API for building and training neural networks with TensorFlow
    • Intended audience: neural network developers
  • ndarray

    • Generic utility library for n-dimensional data I/O operations
    • Used by TensorFlow but does not depend on TensorFlow
    • Intended audience: any developer who needs a Java n-dimensional array implementation, whether or not they use it with TensorFlow

Communication

This repository is maintained by TensorFlow JVM Special Interest Group (SIG). You can easily join the group by subscribing to the jvm@tensorflow.org mailing list, or you can simply send pull requests and raise issues to this repository. There is also a sig-jvm Gitter channel.