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: