Building ML framework with Rust and Category Theory