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 marmaduke on Friday September 29, 2023 @01:49AM (#37699738)
If you like functional languages, then Jax will fit better for you. It provides a bunch of function transformations to implement eg grad, JIT etc.
[ Parent | on HN ]

There are no comments on this comment yet.