JAX: Differentiable Computing by Google

Open in new window