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
相关
- 项目
- 项目
- 项目
- 项目
- 项目