pytorch

Tags: ml, python

Torchscript

• source: https://github.com/pytorch/pytorch/blob/master/torch/script.h

rust torchscript

• https://github.com/LaurentMazare/tch-rs
  • https://github.com/LaurentMazare/tch-rs/blob/main/examples/pretrained-models/main.rs
  • example: https://github.com/LaurentMazare/tch-rs/tree/main/examples/jit
• rust bert: https://github.com/guillaume-be/rust-bert
  • tokenizer: https://github.com/guillaume-be/rust-bert/blob/8ed974b2c7c3490f0632fbc54043d2132a2f6ea9/examples/token_classification.rs#L42a
• rust Sentence Transformer: https://github.com/mladvladimir/rust-sentence-transformers/blob/master/src/model/bert.rs
  • also https://github.com/cpcdoy/rust-sbert