[emu]

We (JAX) see JAX and Numba as mostly complementary; they have different strengths and they are focused on different things.

We haven't tried combining them yet, but we think it would be fun to explore (https://github.com/google/jax/issues/1870). For example, you could use Numba to hand write a numerical kernel that then participates in a machine learning model that uses JAX automatic differentiation.