ParentFull threaddevetec·Nope, probably not. It's using JAX, which works on both GPUs and TPUs.View on HN