top | item 39876393

(no title)

drdirk | 1 year ago

Jax uses the XLA compiler which is compatible with GPU and CPU.

discuss

order

ein0p|1 year ago

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?