search

anowell / are-we-learning-yet

How ready is Rust for Machine Learning?
http://arewelearningyet.com
Creative Commons Attribution 4.0 International
441 stars 61 forks source link

Add the tch crate. #41

Closed LaurentMazare closed 5 years ago

LaurentMazare commented 5 years ago

This adds the tch crate which provides bindings for PyTorch for rust. The bindings are advanced enough so that you can train a LSTM on text data or a ResNet on some image dataset.

anowell commented 5 years ago

Thanks! Merged and should be live in a minute or two.