pyro-ppl/pyro

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

Ouverte

#2 630 ouverte le 16 sept. 2020

 (2 commentaires) (0 réaction) (0 personne assignée)Python (981 forks)batch import
help wantedrefactor

Métriques du dépôt

Stars
 (8 211 étoiles)
Métriques de merge PR
 (Merge moyen 10j 19h) (1 PR mergée en 30 j)

Description

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()

Guide contributeur