Anyone able to install trax on an Apple Silicon, M1 mac?

Course Q&A Natural Language Processing NLP with Sequence Models
7

I got jaxand jaxlib on my M1 mac but no joy with tensorflow-text or trax.

UPDATE: got tensorflow-text installed in a python 3.8 virtual env

tensorflow-deps           2.8.0                         0    apple
tensorflow-hub            0.12.0                   pypi_0    pypi
tensorflow-macos          2.8.0                    pypi_0    pypi
tensorflow-metal          0.4.0                    pypi_0    pypi
tensorflow-text           2.8.1                    pypi_0    pypi

still no joy on trax