machadoprx/Pure-JAX-Transformer not found