Boosting Machine Learning Performance With Rust

Experimenting with Convolutional Neural Networks (CNNs) from scratch in Rust. In my previous article (Part 1) I started my experiment to develop a machine learning framework in Rust from scratch. The main aim of my experiment was to gauge model training speed improvements that can be attained by using Rust in conjunction with PyTorch over a Python equivalent. Results were very encouraging for Feedforward Networks. In this article I continue building on that, with the main objective being to be able to define and train Convolutional Neural Networks (CNNs). As in the previous article, I continue to make use of the Tch-rs Rust crate as a wrapper to the PyTorch C++ Library LibTorch, primarily to access the tensors linear algebra and autograd functions, and the rest is developed from scratch. The code for Part 1 and 2 are now available on Github <a href="https://medium.com/@vincevella/boosting-machine-learning-performance-with-rust-part-2-5e4c2d0adf19">Visit Now</a>