xla

fb29a49e - type hint for xla_model (#7836)