efax:JAX的指数族 源码

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