google/jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

What it solves

JAXoptは、ハードウェア加速、バッチ処理可能、かつ微分可能なオプティマイザのライブラリを提供します。GPUやTPUを使用して最適化問題を解決し、それらの問題の解を、入力に対して微分することが可能です。

How it works

JAXの上に構築されており、このライブラリはJAXの vmap を活用して自動ベクトル化を行い、同一の最適化問題の複数のインスタンスを同時に処理します。最適化解の微分は、陰的微分(implicit differentiation)または展開されたアルゴリズムの反復による自動微分(autodiff)のいずれかを通じてサポートされています。

Who it's for

微分可能なプログラミング・ワークフローに統合できる高性能な最適化ツールを必要とする、JAXを使用している研究者や開発者です。

Highlights

  • CPU、GPU、およびTPUのハードウェア加速。
  • JAXの vmap による自動ベクトル化で、最適化問題のバッチ処理が可能。
  • 最適化解の陰的微分と展開された微分、両方をサポート。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト