This document is the API specification for the Deep Java Library (DJL) PyTorch Engine.
The PyTorch Engine module contains the PyTorch implementation of the DJL EngineProvider. See here for more details.
| Package | Description |
|---|---|
| ai.djl.pytorch.engine |
Contains classes to interface with the underlying PyTorch Engine.
|
| ai.djl.pytorch.jni |
Contains classes to interface with the underlying PyTorch Engine.
|