ML embedding generation module using DJL and PyTorch for the Pipestream platform