Обусловленность элементами матрицы в функции JIT-ted

#python #jax Вопрос: У меня есть функция, которая выглядит следующим образом @jax.jit def f(R): tr = jnp.trace(R) r00 = R[0, 0] r01 = R[0, 1] r02 = R[0, 2] r10…