jax-ml/jax

Name: jax-ml/jax, explained in a minute
Uploaded: 2026-09-28T04:16:41.166Z
Duration: 58 s
Description: You've written a loss function in NumPy, and now you need its gradient, running fast on a GPU, for every example in a batch.

explained in about a minute