Skip to content
Snippets Groups Projects

MNT: change bilby MCMC to use glasflow instead of nflows

6 files
+ 31
32
Compare changes
  • Side-by-side
  • Inline
Files
6
+ 6
6
import torch
import torch
from nflows.distributions.normal import StandardNormal
from glasflow.nflows.distributions.normal import StandardNormal
from nflows.flows.base import Flow
from glasflow.nflows.flows.base import Flow
from nflows.nn import nets as nets
from glasflow.nflows.nn import nets as nets
from nflows.transforms import (
from glasflow.nflows.transforms import (
CompositeTransform,
CompositeTransform,
MaskedAffineAutoregressiveTransform,
MaskedAffineAutoregressiveTransform,
RandomPermutation,
RandomPermutation,
)
)
from nflows.transforms.coupling import (
from glasflow.nflows.transforms.coupling import (
AdditiveCouplingTransform,
AdditiveCouplingTransform,
AffineCouplingTransform,
AffineCouplingTransform,
)
)
from nflows.transforms.normalization import BatchNorm
from glasflow.nflows.transforms.normalization import BatchNorm
from torch.nn import functional as F
from torch.nn import functional as F
# Turn off parallelism
# Turn off parallelism
Loading