If you are looking for more performance and type safety versus python, PyTorch does have a C++ frontend. https://pytorch.org/cppdocs/frontend.html#end-to-end-e

If you are looking for more performance and type safety versus python, PyTorch does have a C++ frontend. https://pytorch.org/cppdocs/frontend.html#end-to-end-example https://pytorch.org/cppdocs/frontend.html#end-to-end-example