RustyTorch is a project aimed at trying and implementing a deep learning framework in Rust. The goal is to leverage Rust's performance and safety features to create a robust and efficient machine learning library.
-
Main Module
main.rs
: Entry point of the application.
-
Storage Module
mod.rs
: Defines theStorage
struct and its implementation.utils.rs
: Contains utility types and enums such asDataPtr
,DataType
, andDevice
.
To get started with RustyTorch, clone the repository and follow the instructions in the Cargo.toml
file to set up the dependencies.
Contributions are welcome! Please refer to the CONTRIBUTING.md
file for guidelines on how to contribute to this project.
This project is licensed under the MIT License. See the LICENSE
file for more details.