Skip to content

DJL - Model Zoo

Deep Java Library's (DJL) Model Zoo is more than a collection of pre-trained models. It's a bridge between a model vendor and a consumer. It provides a framework for developers to create and publish their own models.

A ZooModel has the following characteristics:

  • Globally unique: similar to Java maven packages, a model has its own group ID and artifact ID that uniquely identify it.
  • Versioned: the model version scheme allows developers to continuously update their model without causing a backward compatibility issue.
  • Ready to use out of box: the model contains predefined pre-process and post-process functionality, which allows the user to run inference with a plain java object.
  • Can be published anywhere: models can be published to an S3 bucket, a web server, or a local folder.

Basic model zoo

We provide engine agnostic ZooModels in basic model zoo package. They can be used on any DJL backend engine.

MXNet symbolic model zoo

Apache MXNet has a large number of existing pre-trained models. We created an Apache MXNet model zoo to make it easy for users to consume them.

PyTorch model zoo

We created a PyTorch model zoo to make it easy for users to consume them.

TensorFlow model zoo

We created an TensorFlow model zoo to make it easy for users to consume them.

Publish your own model to the model zoo

You can create your own model in the model zoo so customers can easily consume it. For more information, see Add a new Model to the model zoo .

Load models from ModelZoo

See: How to load model