jax

bde是一个Python包，通过MILE方法将基于采样的贝叶斯深度学习带给实践者，结合了JAX的速度和scikit-learn的API，用于表格监督学习任务。

CAX是一个高性能的开源库，用于细胞自动机研究，基于JAX构建。它可以将模拟加速高达2000倍，并支持任意维度的离散和连续自动机。