Can-t Stop till you get enough: rewriting Pytorch in Rust

172 points marsmute 8 comments

Hey r/rust

I am working on loose rewrite of pytorch into rust. I wanted to start sharing what I am learned and what you should think about if you want to do it yourself!

I gotten to getting gpt-2 loaded and training and I am working towards gpt-oss

If you think this is fun project there are some issues for people to grab!

I am working towards getting a 0.0.1 version ready to put up on crates.io soon!