✳flâneur — a map of the web's best reading

The Sharp Bits 🔪 — JAX documentation

jax.readthedocs.io · 4,526 words · saved by 1 readers

When walking about the countryside of Italy, the people will not hesitate to tell you that JAX has “una anima di pura programmazione funzionale”.

🔪 JAX - The Sharp Bits 🔪 # When walking about the countryside of Italy, the people will not hesitate to tell you that JAX has “una anima di pura programmazione funzionale” . JAX is a language for expressing and composing transformations of numerical programs. JAX is also able to compile numerical programs for CPU or accelerators (GPU/TPU). JAX works great for many numerical and scientific programs, but only if they are written with certain constraints that we describe below. import numpy as np from jax import jit from jax import lax from jax import random import jax import jax.numpy as jnp…

Explore this link on the map →

related reading