Skip to content

DJL TensorFlow Engine

This directory contains the Deep Java Library (DJL) EngineProvider for TensorFlow.

It is based off the TensorFlow Deep Learning Framework.

The DJL TensorFlow Engine allows you to run prediction with TensorFlow or Keras models using Java. Refer to How to import TensorFlow models for loading TF models in DJL.

Modules

  • TensorFlow core api: the TensorFlow 2.x java binding.
  • TensorFlow engine: TensorFlow engine adapter for DJL high level APIs. (NDArray, Model, Predictor, etc)
  • TensorFlow model zoo: Includes pre-trained TensorFlow models and built-int translators for direct import and use.
  • TensorFlow native library: A placeholder to automatically detect your platform and download the correct native TensorFlow libraries for you.