tfjs-npy | NumPy file parsing and serialization for TensorFlow.js | Parser library

I wonder if it is possible, to save and load tensors in tensorflow.js in order to avoid recalculating them for each batch? The problem is that my gpu is barely used because it has to wait for cpu transforming my array to tensor, before the training.

my worflow now looks like this:

1. loading dataset(reading from hdd to array) (1-2 seconds)

2.cpu transforming array to tensor (takes a long time)

3.gpu trains (takes 1 second or less)

1. unloading / tidy (5 seconds, also a bit too long)

2. repeat

EDIT: Here is some code with the problematic(means long heavy computation) and unproblematic lines commented: