Hi, I installed it as a python package and the code runs super slow on 3D stack, do you have an idea how to speed it up? Thanks