Swift for TensorFlow – A system for deep learning and differentiable computing

Re:Swift for TensorFlow – A system for deep learning and dif... by anoncept on Sunday September 20, 2020 @11:16AM (#24534492)

As a starting point, maybe take a look at the 'tch' crate [1] and the 'rust-bert' crate [2] built on top of it?

[1]: https://github.com/LaurentMazare/tch-rs -> https://crates.io/crates/tch

[2]: https://github.com/guillaume-be/rust-bert -> https://crates.io/crates/rust-bert

[ Parent | on HN ]

Re:Swift for TensorFlow – A system for deep learning and dif... by danieldk on Sunday September 20, 2020 @01:58PM (#24535809)

tch-rs is really nice. I have built a Rust sequence labeler + dependency parser + lemmatizer on top of it. It supports multi-task learning, finetuning of BERT-like models, model distillation, etc.:

https://github.com/stickeritis/sticker2/

Unfortunately, when I started this guillaume-be's rust-bert crate wasn't around yet (I think), so I also ported some of the models from Huggingface transformers:

https://github.com/stickeritis/sticker-transformers/

At any rate, I can highly recommend the tch crate if you are looking to build neural networks in Rust.

[ Parent | on HN ]