JAX – NumPy on the CPU, GPU, and TPU

Submitted ⁨⁨2⁩ ⁨years⁩ ago⁩ by ⁨bot@lemmy.smeargle.fans [bot]⁩ to ⁨hackernews@lemmy.smeargle.fans⁩

https://jax.readthedocs.io/en/latest/notebooks/quickstart.html