Преобразование функции — операция, которая принимает функцию на вход и возвращает измененную функцию на выходе. В JAX и связанных системах преобразования функций поддерживают такие возможности, как автоматическое дифференцирование, векторизация, компиляция и параллелизация.