thomaspinder/GPJax

解決的問題

GPJax 為實現高斯過程 (GP) 模型提供了一個靈活的低層介面。其目標是彌合紙面上的數學符號與實際代碼之間的鴻溝，允許研究人員在不受可能具有限制的高層抽象約束的情況下，實現並擴展 GP 模型。

工作原理

該函式庫構建在 JAX 之上，利用 JAX 的自動微分和硬體加速來實現各種 GP 方法。它提供了一種定義核函數和執行推理的結構化方法，支持包括共軛推理、稀疏變分推理和拉普拉斯近似在內的廣泛技術。

適用對象

它主要為從事高斯過程研究的研究人員設計，這些研究人員需要最大限度的靈活性來定制他們的模型、創建自定義核函數並嘗試新的推理方法。

亮點

• 廣泛的方法支持：支持回歸、分類和泊松回歸，以及異方差推理和深度核回歸。
• 靈活的定制：允許創建自定義核函數以及在非歐幾里得空間或圖上進行推理。
• JAX 驅動：利用 JAX 進行高效計算和硬體加速。
• 多樣化的推理技術：包括稀疏變分推理、隨機變分推理和拉普拉斯近似的實現。