Обложка канала

Spark in me - Internet, data science, math, deep learning, philosophy

2440 @snakers4

Канал про интересные мне темы - интернет - статистика - наука о данных Без рекламы и буллшита.

Spark in me - Internet, data science, math, deep learning, philosophy

4 года назад
Открыть в
Я долго ходил вокруг jax-а, что-то про него читал и смотрел, но наконец-то более-менее добрался и полноценно потыкал. Мне понравилось, но я подумал, что в продакшн его запускать рановато и написал про это. #Миша #тулинг
jax: почему это круто и почему он вам (скорее всего) не нужен

Как читать этот пост? Если вы уже знаете что такое jax и haiku/flax/optax - можно сразу прыгать до секции "почему плохо для продакшена". Если интересно про опыт использования библиотек, то в секцию "про опыт использования". Если хочется TL;DR, то jax быстрый, на нём приятно писать, у него круто хэндлится рандом, но в нём легко набаговать, если вы не сталкивались с jit-компиляцией и функциональными языками. Если дать jax джуну, то он сможет найти миллион неожиданных способов выстрелить себе в ногу. Документация…

Telegraph