jax

7c8508e5 - Add link to XLA documentation for building JAX with CUDA from sources.