7 ms·Something nice about this guide is that it generally transfers to GPU directly thanks to JAX/XLA.by jdeaton 2y agoSomething nice about this guide is that it generally transfers to GPU directly thanks to JAX/XLA.