conda create -n myenv
conda activate myenv
conda install -c apple tensorflow-deps
pip install tensorflow-macos==2.9.0
pip install jaxlib jax matplotlib optax pytest flax
pip install tensorflow-probability==0.17
conda create -n myenv
conda activate myenv
conda install -c apple tensorflow-deps
pip install tensorflow-macos==2.9.0
pip install jaxlib jax matplotlib optax pytest flax
pip install tensorflow-probability==0.17