使用 NumPy 和 JAX 的概率编程技术,实现自动求导和 JIT 编译至 GPU/TPU/CPU。
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.