@mark_jimenez Thanks.  We have teams working on enabling on-deivce training, but I’m not sure if that can be released in 1.5.0.  As for the API wrappers, we have a proposal internally - https://github.com/pytorch/pytorch/pull/25541 4. We’ve been proactively collecting feedbacks from communities, but haven;t decided when to release it, so feel free to submit ideas, proposals, etc. The 1.5.0 version in .podspec is just a placeholder, nothing particular has been done on that branch.