/jax_examples

Primary LanguageLLVM

JAX Example

The goal is to test if XLA fusion is working correctly.