jaxns:JAX中的嵌套采样 源码

它是什么? 使用嵌套采样启用概率编程。 它以JAX编码的方式允许将整个推理算法降低到XLA原语,而XLA原语是JIT编译的,以实现高性能。 您可以在这里阅读有关内容:( ) JAXNS提供了一种受约束的似然采样器,该采样器结合并修改了MultiNest(F.Feroz等人2008; )和PolyChord(WJ Handley等人2015; )。 通过将sampler_name=slice设置为切片采样,将sampler_name=multi_ellipsoid为拒绝采样,可以提供两个采样器。 安装 确保您拥有JAX并通过pip install jax jaxlib numpy matplotlib scipy了通常的可疑程序pip install jax jaxlib numpy matplotlib scipy 。 使用pip install jaxns或pip instal