4 ms·

If you want type safety on top of Pytorch, checkout this project https://github.com/NVIDIA/NeMo https://github.com/NVIDIA/NeMo

• MiroF 7y ago
  It'd be nice to have the same pytorch abstractions but in a more statically typed language than pytorch.
  • jhj 7y ago
    there's a C++14 frontend https://pytorch.org/cppdocs/frontend.html#end-to-end-example https://pytorch.org/cppdocs/frontend.html#end-to-end-example though the typing doesn't dive into tensor dimensions etc.
  • xvilka 7y ago
    You can use PyTorch Rust bindings [1] or OCaml bindings [2]. [1] https://github.com/LaurentMazare/tch-rs https://github.com/LaurentMazare/tch-rs [2] https://github.com/LaurentMazare/ocaml-torch https://github.com/LaurentMazare/ocaml-torch