After considerable effort I was able to get tensorflow-text installed back when I took the class, but I don’t believe that anyone has ever released a trax version for the M1.