Feature/jax numba mix#187

Merged

Jammy2211 merged 17 commits intofeature/jax_wrapperfrom

feature/jax_numba_mix

Oct 6, 2025

Commits on Oct 1, 2025

moved first 3 numba functions

Jammy2211
authored and
Jammy2211
committed
moved more functions

Jammy2211
authored and
Jammy2211
committed
update dataset w tilde to use numba util

Jammy2211
authored and
Jammy2211
committed
update w_tilde to use numba utils

Jammy2211
authored and
Jammy2211
committed
fix _data_vector_x1_mapper

Jammy2211
authored and
Jammy2211
committed
more direct split of numba code

Jammy2211
authored and
Jammy2211
committed
fix unitt est due to jax operated mapping matrix

Jammy2211
authored and
Jammy2211
committed
test__data_linear_func_matrix_dict

Jammy2211
authored and
Jammy2211
committed
update data_linear_func_matrix_from to not use numba

Jammy2211
authored and
Jammy2211
committed
remove use of frames in off diag

Jammy2211
authored and
Jammy2211
committed
docstring

Jammy2211
authored and
Jammy2211
committed
remove convolver

Jammy2211
authored and
Jammy2211
committed
split mappeR_util with numba

Jammy2211
authored and
Jammy2211
committed
split mesh_util into numba

Jammy2211
authored and
Jammy2211
committed
replace more unit tests so they use numba util

Jammy2211
authored and
Jammy2211
committed
fix mesh numab improt

Jammy2211
authored and
Jammy2211
committed

Commits on Oct 6, 2025

JAX complet

Jammy2211
authored and
Jammy2211
committed