こんにちは、みなさん！
Pythonプロジェクト Parax を共有したくて投稿しました。これは Equinox ライブラリの上に構築したアドオンで、JAXにおける「パラメータ優先」のモデリングに対応しています。
私たちの科学アプリケーションでは、固定としてマークしたり、事前確率分布を紐づけたりといった具合に、パラメータオブジェクトにメタデータを付与する必要があることがよくありました。さらに、非常に深い階層の中でこれらのパラメータを操作する必要も多く、状況によっては eqx.tree_at を使うと直感的でないことがあります。
そこで、Paraxを開発しました。これはparax.Parameter と parax.Module（どちらも eqx.Module を継承）に加えて、いくつかの補助ユーティリティを提供します。これにより、Equinoxの不変（immutable）の原則に従いながらも、よりオブジェクト指向的なモデルの検査・操作のアプローチが可能になります。
いくつかの ドキュメント と、いくつかの例もあります。もしかすると、どなたか他の方にとって役に立つかもしれません！ :)
それでは、
Gary
[link] [comments]
