A C D F G H I L M N P R S T W
All Classes All Packages

A

addEnv(String, String) - Method in class ai.djl.python.engine.PyEnv

Adds an environment variable.

addParameter(String, String) - Method in class ai.djl.python.engine.PyEnv

Adds a model initialization parameter.

ai.djl.python.engine - package ai.djl.python.engine

Contains classes to interface with the underlying Python worker.

allocateDirect(int) - Method in class ai.djl.python.engine.PyNDManager

C

close() - Method in class ai.djl.python.engine.PyModel

close() - Method in class ai.djl.python.engine.PyNDManager

CodecUtils - Class in ai.djl.python.engine

This is a utility class for reading and writing to netty ByteBuf.

create(Buffer, Shape, DataType) - Method in class ai.djl.python.engine.PyNDManager

D

detach() - Method in class ai.djl.python.engine.PyNDArray

DsEngineProvider - Class in ai.djl.python.engine

DsEngineProvider is the DeepSpeed implementation of EngineProvider.

DsEngineProvider() - Constructor for class ai.djl.python.engine.DsEngineProvider

Constructs a new DsEngineProvider instance.

DsNDManager - Class in ai.djl.python.engine

DsNDManager is the Python engine implementation of NDManager.

F

from(NDArray) - Method in class ai.djl.python.engine.PyNDManager

G

getAlternativeEngine() - Method in class ai.djl.python.engine.PyEngine

getEngine() - Method in class ai.djl.python.engine.DsNDManager

getEngine() - Method in class ai.djl.python.engine.PyEngineProvider

getEngine() - Method in class ai.djl.python.engine.PyNDManager

getEngineName() - Method in class ai.djl.python.engine.DsEngineProvider

getEngineName() - Method in class ai.djl.python.engine.PyEngine

getEngineName() - Method in class ai.djl.python.engine.PyEngineProvider

getEngineRank() - Method in class ai.djl.python.engine.DsEngineProvider

getEngineRank() - Method in class ai.djl.python.engine.PyEngineProvider

getEntryPoint() - Method in class ai.djl.python.engine.PyEnv

Returns the model's entrypoint file path.

getHandler() - Method in class ai.djl.python.engine.PyEnv

Returns the python model's handler function.

getInitParameters() - Method in class ai.djl.python.engine.PyEnv

Returns the python model initialization parameters.

getModelLoadingTimeout() - Method in class ai.djl.python.engine.PyEnv

Returns the model loading timeout in seconds.

getPredictTimeout() - Method in class ai.djl.python.engine.PyEnv

Returns the prediction timeout in seconds.

getPythonExecutable() - Method in class ai.djl.python.engine.PyEnv

Returns the python executable path.

getRank() - Method in class ai.djl.python.engine.PyEngine

getTensorParallelDegree() - Method in class ai.djl.python.engine.PyEnv

Returns the tensor parallel degree.

getVersion() - Method in class ai.djl.python.engine.PyEngine

H

hasCapability(String) - Method in class ai.djl.python.engine.PyEngine

I

installDependency(Path) - Method in class ai.djl.python.engine.PyEnv

Installs model dependencies if needed.

intern(NDArray) - Method in class ai.djl.python.engine.PyNDArray

L

load(Path, String, Map<String, ?>) - Method in class ai.djl.python.engine.PyModel

M

MAX_BUFFER_SIZE - Static variable in class ai.djl.python.engine.CodecUtils

mpiMode - Variable in class ai.djl.python.engine.PyEngineProvider

N

newBaseManager() - Method in class ai.djl.python.engine.PyEngine

newBaseManager(Device) - Method in class ai.djl.python.engine.PyEngine

newGradientCollector() - Method in class ai.djl.python.engine.PyEngine

newModel(String, Device) - Method in class ai.djl.python.engine.PyEngine

newPredictor(Translator<I, O>, Device) - Method in class ai.djl.python.engine.PyModel

newSubManager(Device) - Method in class ai.djl.python.engine.PyNDManager

newSymbolBlock(NDManager) - Method in class ai.djl.python.engine.PyEngine

P

PyEngine - Class in ai.djl.python.engine

The PyEngine is an implementation of the Engine that runs Python worker.

PyEngineProvider - Class in ai.djl.python.engine

PyEngineProvider is the Python implementation of EngineProvider.

PyEngineProvider() - Constructor for class ai.djl.python.engine.PyEngineProvider

PyEnv - Class in ai.djl.python.engine

Python engine environment.

PyEnv(boolean) - Constructor for class ai.djl.python.engine.PyEnv

Constructs a new PyEnv instance.

PyModel - Class in ai.djl.python.engine

PyModel is the Python engine implementation of Model.

PyNDArray - Class in ai.djl.python.engine

PyNDArray is the Python engine of NDArray.

PyNDManager - Class in ai.djl.python.engine

PyNDManager is the Python engine implementation of NDManager.

R

readBytes(ByteBuf, int) - Static method in class ai.djl.python.engine.CodecUtils

Reads the specified length of data.

readUtf8(ByteBuf) - Static method in class ai.djl.python.engine.CodecUtils

Read a String from the ByteBuf.

S

setEntryPoint(String) - Method in class ai.djl.python.engine.PyEnv

Sets the model's entrypoint file path.

setHandler(String) - Method in class ai.djl.python.engine.PyEnv

Sets the python model's handler function.

setModelLoadingTimeout(int) - Method in class ai.djl.python.engine.PyEnv

Sets the model loading timeout in seconds.

setPredictTimeout(int) - Method in class ai.djl.python.engine.PyEnv

Sets the prediction timeout in seconds.

setPythonExecutable(String) - Method in class ai.djl.python.engine.PyEnv

Sets the python executable path.

setRandomSeed(int) - Method in class ai.djl.python.engine.PyEngine

setTensorParallelDegree(int) - Method in class ai.djl.python.engine.PyEnv

Sets the tensor parallel degree.

T

toByteBuffer() - Method in class ai.djl.python.engine.PyNDArray

toString() - Method in class ai.djl.python.engine.PyEngine

W

writeUtf8(ByteBuf, String) - Static method in class ai.djl.python.engine.CodecUtils

Encode a String in UTF-8 and write it to the ByteBuf.

A C D F G H I L M N P R S T W
All Classes All Packages