feat(train): support opt-in $ref loading

feat(train): support opt-in $ref loading #8436