Implement inference module