Some temporary stubs for training and inference modules