cannot install on Apple Silicon

[chunyang-wen]

I am following the doc here:

https://github.com/apple/axlearn/blob/a7a738e816363ff4950b0da391206c3216e87d6d/docs/01-start.md

Failed on the final step:

pip install "axlearn[core, apple-silicon]"

ERROR: Ignored the following versions that require a different python version: 1.6.2 Requires-Python >=3.7,<3.10; 1.6.3 Requires-Python >=3.7,<3.10; 1.7.0 Requires-Python >=3.7,<3.10; 1.7.1 Requires-Python >=3.7,<3.10
ERROR: Could not find a version that satisfies the requirement tensorflow_text==2.8.2 (from axlearn[core,gcp]) (from versions: 2.17.0rc0, 2.17.0, 2.18.0rc0)
ERROR: No matching distribution found for tensorflow_text==2.8.2

MacOS: 15.0.1
Conda env: python 3.10

And when I try to build and install tensorflow_text ==2.6.1, it installs tensorflow==2.16.2. I don't know, but this may be no related.

[markblee]

Does it work if you install from source instead? pip install -e .[core,apple-silicon]. The pypi wheel may be slightly behind.

[chunyang-wen]

ERROR: Ignored the following versions that require a different python version: 1.6.2 Requires-Python >=3.7,<3.10; 1.6.3 Requires-Python >=3.7,<3.10; 1.7.0 Requires-Python >=3.7,<3.10; 1.7.1 Requires-Python >=3.7,<3.10
ERROR: Could not find a version that satisfies the requirement tfds-nightly (from seqio) (from versions: none)
ERROR: No matching distribution found for tfds-nightly