dfm/tinygp
The tiniest of Gaussian Process libraries
解決的問題
提供一種輕量、快速且高效的 Python 高斯過程(GP)模型實作方式,避免重型庫的開銷,同時保持高效率。
工作原理
基於 JAX 建構,利用 JAX 的自動微分與 GPU 加速能力,加速 GP 模型的計算。
適用對象
需要一個輕量但功能強大的高斯過程庫,並能與 JAX 生態系統無縫整合的研究人員與開發者。
特色亮點
- 支援 GPU 加速
- 自動微分
- 輕量且簡潔的介面
- 經由基準測試證明的高效率
相關
- 專案
- 專案
- 專案
- 專案
- 專案