efax:JAX的指数族

EFAX：JAX中的指数族 该库提供了一组用于处理差分编程库中的指数族分布的工具。 指数族是概率分布的重要一类，包括正态分布，伽玛分布，β分布，指数分布，泊松分布，二项分布和伯努利分布。 有关此库背后基本概念的说明，请参见我们的。 框架 表示 EFAX的对象有一个单一的基类：对分布族和分布Parametrization进行编码的参数化。 每个这样的对象都有一个形状，因此它可以存储任意数量的分布。 在此类物体上操作时，将应用NumPy的广播规则。 这与SciPy不同，后者的每个分布都由一个对象表示，因此一千个分布需要一千个对象。 所有参数化对象都是使用tjax.dataclass数据类。 这些数据类是对Python数据类的修改，以支持JAX的类型注册。 这样可以轻松标记静态属性。 在JAX中，静态属性是一种在功能更改时引起函数重新编译的属性，因此，使用该属性可以做什么具有更大的灵活性。