"reescrevê-lo em jax" vai ser o próximo "reescrevê-lo em rust"