Jax is an accelerated linear algebra library that allows for high-performance array computing and automatic differentiation. It enforces constraints like immutable arrays and pure functions. Jax can run on accelerated hardware like GPUs and TPUs.

3m watch time

Sort: