Compatibility Issues with new M-series chip on Mac

[yanbronshtein]

Hello,

It looks like the model validation toolkit is not handling the new apple chips.

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.

[aeftimia]

It looks like for python3.9.6, the latest jax/jaxlib support Apple's new M1 chips. Let me know if switching to that version solves the problem

[yanbronshtein]

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.

For python=3.9.12