Tensorflow Metal Malfunctioning Completely

I am just starting to learn neural networks.

If I run my code and try to fit a simple trigonometric function, the model builds a good-looking function.

If I pip install tensorflow-metal and run, I get a straight line not resembling the non-linear function at all.

if I uninstall metal, everything works again. Which suggests there is something wrong with metal.

Any help would be appreciated. I would use the metal acceleration for the next steps in my project.

Thank you