M1 MacOS support

[gnsoares]

Hello,

The tensforflow package cannot be installed in M1 Macs, so I propose modifying this line in the setup.py file from:

'tensorflow==2.11.1'

to:

 'tensorflow==2.11.1' if sys.platform != 'darwin' or platform.processor() != 'arm' else 'tensorflow-macos==2.11.0',

[uptacks]

I have been able to get this running on the MBA M2 2022. This can be done by commenting out the

'tensorflow==2.11.1'

line and replacing it to:

'tensorflow-macos',
'tensorflow-metal',

instead. This likely should be treated as a temporary fix to get up and running.

[duenez]

Has this been resolved? Can we close this?

[dhrumilp15]

Running into RuntimeError: This version of jaxlib was built using AVX instructions, which your CPU and/or operating system do not support. You may be able work around this issue by building jaxlib from source.