numpyro

# 导入所需的函数和类 from numpyro.contrib.funsor.enum_messenger import trace as enum_tr from numpyro.contrib.funsor.enum_messenger import enum # 使用 numpyro.han ......