The way you've abstracted the linear layer to enable easy extension to other types of layers, such as CNN or LSTM, is a clever design choice. It makes your framework much more versatile and future-proof. I'm interested to know if you've encountered any challenges when implementing this design in Rust compared to Python. Great write-up and experiment!