Added BURN library to Cargo

README.md

@@ -1,7 +1,9 @@
-<h1 align="center">Dockerized-Rust</h1>
+<h1 align="center">Dockerized-Rust-ML</h1>
-<p align="center">A Template for Rust Projects</p>
+<p align="center">A Template for Rust ML Projects</p>
 
 ## About this Template
-This template should be used when developing **Rust** apps. It's designed to be *lightweight* and *Fast*. I would recommend using this **with** my **Rust-LS** container as it makes Rust Development so much more fun and streamlined
+This template should be used when developing **Rust** apps. It's designed to be *lightweight* and *Fast*. This project is built on top of my **Dockerized-Rust** template that more fits towards *Machine Learning*. This project is still in early infancy so be prepared for bugs and issues
+
+I would recommend using this **with** my **Rust-LS** container as it makes Rust Development so much more fun and streamlined
 
 Just adopt this into your project and it should take care of itself!

project/Cargo.toml

@@ -4,3 +4,4 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+burn = { version = "0.16.0", features = ["webgpu", "train", "vision"] }