jax

Вопрос или проблема Я пытаюсь использовать custom_vjp из JAX для определения пользовательских вычислений градиентов для функции, которая принимает выражение SymPy в качестве входного параметра. Однако я сталкиваюсь с ошибками, потому что JAX не поддерживает

Вопрос или проблема У меня есть функция, которая возвращает список массивов, и мне нужно найти ее производную по одному параметру. Например, пусть у нас есть def fun(x): … return [a,b,c] где a, b, c и d — это многомерные массивы (например, вещественные