rework OmniTorch - init