@a_d
I don’t think we support amp in libtorch at this moment. @mcarilli
You can find some workaround in this post.
‘Deploy mixed precision model in libtorch 33’
I think Apex is a mix of its own libs and pytorch libs, not easy to use it purely from C++ I guess.