08.11.2022, Мелешенко Иван, г. Самара Ускорение вычислений при помощи фреймворка машинного обучения JAX
В процессе написания научной работы, я столкнулся с такой проблемой, как относительно невысокая скорость выполнения вычислений. Из-за этого приходится тратить больше времени или жертвовать точностью вычислений, но что делать, если не хочется идти на компромисс и чем-то жертвовать? Воспользоваться новым фреймворком JAX от google. В связке с различными ускорителями он покажет достойный результат.