News for hackers. Stuff that matters.

A comment on JAX – NumPy on the CPU, GPU, and TPU, with great automatic differentiation. Parent comment

Re:JAX – NumPy on the CPU, GPU, and TPU, with great automati... by matrss on Friday September 29, 2023 @04:50AM (#37701000)
Meanwhile, the first sentence in their readme is this:

> JAX is Autograd and XLA, brought together for high-performance machine learning research.

That does not really convey the generality of it that well.

[ Parent | on HN ]
Re:JAX – NumPy on the CPU, GPU, and TPU, with great automati... by mattjjatgoogle on Friday September 29, 2023 @12:00PM (#37706167)
You're right! Maybe we should revise that... I made https://github.com/google/jax/pull/17851, comments welcome!
[ Parent | on HN ]