with_cpu
Move MSE to separate function
Activation functions implementaions
Fix training issues
AvgPool2d implementation
Make test convnet cpp app
revert compiler compatibility to gcc12
Variable constructor refactoring + ReLU+ LeakyReLU
BatchNorm2d module interface and tensor scalar filling
Add tests for Linear layer and update matmul usage
MaxPool2d implementation