dfm/tinygp

The tiniest of Gaussian Process libraries

解决的问题

提供一种轻量、快速且高效的 Python 高斯过程(GP)模型实现方式,避免了重型库的开销,同时保持高性能。

工作原理

基于 JAX 构建,利用 JAX 的自动微分和 GPU 加速能力,加速 GP 模型的计算。

适用人群

需要一个轻量但功能强大的高斯过程库,并能无缝集成到 JAX 生态系统中的研究人员和开发者。

特色亮点

  • 支持 GPU 加速
  • 自动微分
  • 轻量且简洁的接口
  • 通过基准测试证明的高性能

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目