jax-ml/oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
解決的問題
Oryx 提供了一個用於機率程式設計與深度學習的框架，讓使用者能夠建構複雜的統計模型，並將其與 JAX 的高效能運算能力整合。
運作方式
它實作了一組函數轉換功能，旨在與 JAX 的核心轉換功能無縫組合與整合，例如 jit（即時編譯）、grad（自動微分）和 vmap（向量化）。
適用對象
需要將機率模型與 JAX 的速度和彈性結合的研究人員與開發者。
特色
- 建立在 JAX 之上
- 支援機率程式設計
- 與 JAX 轉換功能整合，例如
jit、grad和vmap
相關
- 專案
- 專案
- 專案
- 專案
- 專案