JAXによるスケーラブルな機械学習 - ZOZO TECH BLOG

はじめに こんにちは、ZOZO NEXT ZOZO ResearchのSai Htaung Khamです。ZOZO NEXTは、ファッション領域におけるユーザーの課題を想像しテクノロジーの力で解決すること、より多くの人がファッションを楽しめる世界の創造を目指す企業です。 ZOZO NEXTでは多くのアルゴリズムを研究開発しており、その中でJAXというライブラリを使用しています。JAXは高性能な機械学習のために設計されたPythonのライブラリです。NumPyに似ていますが、より強力なライブラリであると考えることができます。NumPyとは異なり、JAXはマルチGPU、マルチTPU、そして機械学習の研究に非常に有用な自動微分（Autograd）をサポートしています。 JAXはNumPyのAPIのほとんどをミラーリングしているので、NumPyライブラリに慣れている人なら非常に導入しやすいです。A

[mohritaroh]

（本論に至る全然まえのはなしだけど、TPUってTensor Processing Unitの略なのか、機械学習に特化したプロセッサ）

[mysql8]

flaxしか勝たん

[misshiki]

“実世界のデータを使った機械学習を、JAXライブラリで実行する方法について説明”

[sonots]

お、JAX事例だ