01Entry
JAXNS Sampling
Probabilistic programming framework using nested sampling as the inference engine, implemented in JAX. The entire algorithm lowers to XLA primitives and JIT-compiles for high performance in scientific machine learning.
01Entry
Probabilistic programming framework using nested sampling as the inference engine, implemented in JAX. The entire algorithm lowers to XLA primitives and JIT-compiles for high performance in scientific machine learning.