mlx conversion for MLP (in Unet.py) and test

the MLX conversion for MLP and the corresponding test of mlx v PyTorch.