Path	Lines of Code
spinoffs/oryx/oryx/__init__.py	7
spinoffs/oryx/oryx/bijectors/__init__.py	12
spinoffs/oryx/oryx/bijectors/bijector_extensions.py	113
spinoffs/oryx/oryx/core/__init__.py	28
spinoffs/oryx/oryx/core/interpreters/__init__.py	5
spinoffs/oryx/oryx/core/interpreters/harvest.py	735
spinoffs/oryx/oryx/core/interpreters/inverse/__init__.py	4
spinoffs/oryx/oryx/core/interpreters/inverse/core.py	262
spinoffs/oryx/oryx/core/interpreters/inverse/custom_inverse.py	84
spinoffs/oryx/oryx/core/interpreters/inverse/rules.py	160
spinoffs/oryx/oryx/core/interpreters/inverse/slice.py	74
spinoffs/oryx/oryx/core/interpreters/log_prob.py	72
spinoffs/oryx/oryx/core/interpreters/propagate.py	212
spinoffs/oryx/oryx/core/interpreters/unzip.py	543
spinoffs/oryx/oryx/core/kwargs_util.py	41
spinoffs/oryx/oryx/core/ppl/__init__.py	17
spinoffs/oryx/oryx/core/ppl/effect_handler.py	96
spinoffs/oryx/oryx/core/ppl/plate_util.py	25
spinoffs/oryx/oryx/core/ppl/transformations.py	118
spinoffs/oryx/oryx/core/primitive.py	157
spinoffs/oryx/oryx/core/pytree.py	19
spinoffs/oryx/oryx/core/serialize.py	17
spinoffs/oryx/oryx/core/state/__init__.py	17
spinoffs/oryx/oryx/core/state/api.py	42
spinoffs/oryx/oryx/core/state/function.py	190
spinoffs/oryx/oryx/core/state/module.py	56
spinoffs/oryx/oryx/core/state/registrations.py	70
spinoffs/oryx/oryx/core/trace_util.py	77
spinoffs/oryx/oryx/distributions/__init__.py	8
spinoffs/oryx/oryx/distributions/distribution_extensions.py	85
spinoffs/oryx/oryx/experimental/__init__.py	4
spinoffs/oryx/oryx/experimental/autoconj/addn.py	47
spinoffs/oryx/oryx/experimental/autoconj/canonicalize.py	101
spinoffs/oryx/oryx/experimental/autoconj/einsum.py	94
spinoffs/oryx/oryx/experimental/matching/__init__.py	3
spinoffs/oryx/oryx/experimental/matching/jax_rewrite.py	348
spinoffs/oryx/oryx/experimental/matching/matcher.py	241
spinoffs/oryx/oryx/experimental/matching/rules.py	103
spinoffs/oryx/oryx/experimental/mcmc/__init__.py	7
spinoffs/oryx/oryx/experimental/mcmc/kernels.py	121
spinoffs/oryx/oryx/experimental/mcmc/utils.py	18
spinoffs/oryx/oryx/experimental/nn/__init__.py	21
spinoffs/oryx/oryx/experimental/nn/base.py	265
spinoffs/oryx/oryx/experimental/nn/combinator.py	52
spinoffs/oryx/oryx/experimental/nn/convolution.py	155
spinoffs/oryx/oryx/experimental/nn/core.py	101
spinoffs/oryx/oryx/experimental/nn/function.py	16
spinoffs/oryx/oryx/experimental/nn/normalization.py	103
spinoffs/oryx/oryx/experimental/nn/pooling.py	72
spinoffs/oryx/oryx/experimental/nn/reshape.py	35
spinoffs/oryx/oryx/experimental/optimizers/__init__.py	6
spinoffs/oryx/oryx/experimental/optimizers/optix.py	211
spinoffs/oryx/oryx/internal/__init__.py	1
spinoffs/oryx/oryx/tools/build_oryx_docs.py	28
spinoffs/oryx/oryx/util/__init__.py	2
spinoffs/oryx/oryx/util/summary.py	7
spinoffs/oryx/oryx/version.py	11
spinoffs/oryx/pyproject.toml	31