pyro-ppl/pyro

List: move these things upstream to PyTorch, Funsor, etc.

开放

#2,630 创建于 2020年9月16日

 (2 条评论) (0 个反应) (0 位负责人)Python (981 个派生)batch import
help wantedrefactor

仓库指标

星标
 (8,211 个星标)
PR 合并指标
 (平均合并 10天 19小时) (30 天内合并 1 个 PR)

描述

This issue tracks Pyro components that should be moved to upstream dependencies.

Move to PyTorch

  • constraints.IndependentConstraint
    • usage in MultivariateNormal
    • usage in dist.Independent
  • dist.Delta #1781
  • constraints.integer
  • Geometric.log_prob() as patched in pyro/distributions/torch.py
  • Uniform's ._unbroadcasted_* logic
  • _with_cache() in pyro/infer/reparam/unit_jacobian.py
  • dist.DirichletMultinomial https://github.com/pytorch/pytorch/issues/56030

Move to Funsor

See also https://github.com/pyro-ppl/funsor/issues/362

  • distributions.util.detach()
  • ops.tensor_utils.haar_transform() and .inverse_haar_transform()

贡献者指南