EFAX：JAX中的指数族 该库提供了一组用于处理差分编程库中的指数族分布的工具。 指数族是概率分布的重要一类，包括正态分布，伽玛分布，β分布，指数分布，泊松分布，二项分布和伯努利分布。 有关此库背后基本概念的说明，请参见我们的。 框架 表示 EFAX的对象有一个单一的基类：对分布族和分布Parametrization进行编码的参数化。 每个这样的对象都有一个形状，因此它可以存储任意数量的分布。 在此类物体上操作时，将应用NumPy的广播规则。 这与SciPy不同，后者的每个分布都由一个对象表