Standardizes code formatting and improves efficiency

[ChaitanyaSirivuri]

Applies consistent code formatting for readability and style Replaces torch.no_grad with torch.inference_mode for better efficiency Adds missing test data loader in feedforward neural network Fixes minor typos and removes unnecessary whitespace