News for hackers. Stuff that matters.
Re:How do I get started with Jax on TPU VMs by ein0p on Saturday March 30, 2024 @01:20PM (#39876696)
Yes, but its performance on GPU leaves much to be desired, and 20 times as much research comes out on PyTorch. Would you rather just build on that or laboriously port and debug the models and their weights, losses, dataset readers, training regimes etc etc?

There are no comments on this comment yet.