みなさんこんにちは!
JAXにおける「パラメトリックモデリング」を扱う私のプロジェクト Parax について近況を共有したいと思います。
以前のParaxは科学用途により重点を置いていましたが、その後、あらゆるタイプのJAX作業に役立つツールとなるように一般化しました。現在は、クリーンで拡張可能なAPIに強くフォーカスしているほか、ライブラリが完全に オプトイン であることを保証しています。これは、よりフレームワークのようなアプローチを取っていた以前のバージョンとは対照的です。
Paraxの主な機能:
- メタデータ付きの導出/制約付きパラメータ
- 計算済みのPyTreeと呼び出し可能なパラメータ化
- 固定・有界・確率的なPyTreeおよびパラメータのための抽象インターフェース
- フィルタリングおよび操作ツール
ドキュメントは こちら で公開されており、いくつかの基本的な例もあります。このパッケージが、どこかの誰かの役に立てば嬉しいです!
それでは、
Gary
[リンク] [コメント]




