JAX – NumPy on the CPU, GPU, and TPU, with great automatic differentiation

Re:JAX – NumPy on the CPU, GPU, and TPU, with great automati... by cl3misch on Friday September 29, 2023 @01:01PM (#37707040)

Ah, I think I was unclear. I specifically meant your reference to Equinox, because that seemed to me to be somewhat ML specific.

In general, I very much agree that "autodiff+autoparallel are really useful for any kind of numerical computation". And the use cases are also really common in my bubble. It's just that (imho) most people have not realized this.

[ Parent | on HN ]

Re:JAX – NumPy on the CPU, GPU, and TPU, with great automati... by patrickkidger on Friday September 29, 2023 @03:30PM (#37709056)

Ah right! Actually it's a good point, the Equinox readme/etc do tend to emphasise the ML use cases -- partly this is deliberate (go where the money is)! But I should probably tweak it to emphasise more general parameterised models.

[ Parent | on HN ]