Первый запуск функции с использованием jax требует значительно больше времени, чем последующие вызовы. Это появляется потому, что в первый раз инициализируется эта функция. Сложность в том, что если мы изменим размер переменной в функции, она будет скомпилирована снова.
Я пишу инструмент обратного тестирования с помощью jax, и во время вычислений я использую другой размер. матрицы (из-за разного количества рабочих дней для каждого конкретного месяца). Знаете ли вы, как с этим справиться?
Может быть, мне нужно определить длину переменной, и если текущая матрица меньше, просто заполнить ее NaN?
Подробнее здесь: https://stackoverflow.com/questions/790 ... -variables
