概要
Rexは、拡散モデルのサンプリング、反転、編集、補間タスクに利用可能な、代数的に可逆な指数型Runge-Kuttaソルバーのファミリーです。
Lawsonの指数変換とMcCallum–Foster可逆カップリングを組み合わせることで、浮動小数点誤差を除けば順方向と逆方向で状態が正確に再現される「正確に可逆」という特徴を持ちます。
また、基盤となるRunge-Kuttaスキームの収束次数を継承し、確率流ODEと逆時間SDEの両設定で機能します。
本リポジトリには、リファレンス実装と、論文で用いられた実験スクリプトが含まれており、計算効率と可逆性を重視する拡散モデルの研究者や開発者にとって有用なツールです。
既存のソルバーのドロップイン代替としても機能します。
互換性・特徴
- Python
- GPU必須
- CLI
- PyTorch対応
- 拡散モデル対応
基本情報
| ライセンス | MIT |
| Stars | 8 |
| Forks | 1 |
| カテゴリ | 画像生成 |
| アクティビティ | low |