JAX/Numpyroで遊ぼう！
2020年末よりJAX/numpyroで楽しんでいます。楽しんでいる内容をシェアするための文書を作成中です。随時アップデートしていく予定です。
現在の内容：自動微分・最適化 (jax.experimental.optimizer/JAXopt)・HMC-NUTS (numpyro)・ガウス過程/flax
PDF版：機械学習で楽しむJAX/numpyro (0.3)
github: jupyter notebooks
本稿で用いているパッケージはこれらです：
JAX/Numpyro/arviz
JAX 101