dm-haikuをよむ - Qiita

はじめに あけましておめでとうございます。今日から2021年です。新年ということでですね、haikuをよんでいこうと思います。 haiku（dm-haiku）とは haiku（dm-haiku）はdeepmind製のjax向け深層学習モデル記述ライブラリです。まだ開発途中でβ版として公開されています。 sonnetをご存じの方には、haikuはsonnetのjax版と言うと分かりやすいと思います。sonnet → sonnet2 → haiku と進化する過程を見ると面白いです。 先月に、DeepMindのブログで、DeepMindがjaxを使っているよ、色々ライブラリ作って公開しているよ、という旨の内容が紹介されていました。また、ODE-GAN（論文、コード）という研究にhaikuが使われています（この論文もDeepMindの方々が執筆）。ODE-GANは、ルンゲクッタ法や、パラメータの

[sh19910711]

"haiku: jax向け深層学習モデル記述ライブラリ / sonnet → sonnet2 → haiku と進化する過程を見ると面白い / ODE-GAN: ルンゲクッタ法やパラメータの勾配に対する正則化のように柔軟な勾配計算を要求 > tensorflowで書くと大変そう"