G github.com JAX: Numpy with Gradients, GPUs and TPUs 132 points one-more-minute 8 years ago 29 comments