Пишите код на NumPy привычным способом, а затем запускайте его на GPU и TPU. Автоматическое дифференцирование и пакетная обработка данных будут выполняться без дополнительных усилий с вашей стороны. - Автоматически вычисляйте производные обычных функций Python и NumPy. - Компилируйте чистые функции с помощью jit для выполнения на аппаратных ускорителях. - Комбинируйте grad, jit и vmap в любом порядке и масштабируйте вычисления на несколько устройств. https://github.com/jax-ml/jax 📱 @git_developer