|
|
|
|
|
|
by emu
2261 days ago
|
|
|
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. |
|
|