ResNet MLX

## Key Architectural Changes - **Unsqueeze Process** Adjusted tensor dimension expansion logic to ensure proper shape compatibility between temporal embeddings and spatial feature maps. - **Temporal Embedding Broadcast** Modified broadcasting logic to align temporal embeddings with spatial dimensions without introducing shape mismatches. --- ## Testing & Validation - **Test Strategy** Implemented straightforward test cases to verify: - Layer behavior consistency - Numerical stability across resolutions - **Resolved Issues** - Fixed **GroupNorm portability** issues related to: - Parameter initialization - Axis/dimension handling during inference - Validated edge cases for temporal embedding integration --- <img width="1470" alt="Screenshot 2025-02-26 at 17 37 18" src="https://github.com/user-attachments/assets/d3d58489-560e-4ced-867a-6a0567bd575d" />