Comment Comment by axpy906 If you’re doing research JAX makes some sense. Probably some Google bias in there too. parent axpy906 2 years ago