README.md
setup.py
oryx/__init__.py
oryx/version.py
oryx.egg-info/PKG-INFO
oryx.egg-info/SOURCES.txt
oryx.egg-info/dependency_links.txt
oryx.egg-info/not-zip-safe
oryx.egg-info/requires.txt
oryx.egg-info/top_level.txt
oryx/bijectors/__init__.py
oryx/bijectors/bijector_extensions.py
oryx/bijectors/bijector_extensions_test.py
oryx/core/__init__.py
oryx/core/kwargs_util.py
oryx/core/kwargs_util_test.py
oryx/core/primitive.py
oryx/core/pytree.py
oryx/core/serialize.py
oryx/core/serialize_test.py
oryx/core/trace_util.py
oryx/core/interpreters/__init__.py
oryx/core/interpreters/harvest.py
oryx/core/interpreters/harvest_test.py
oryx/core/interpreters/inverse.py
oryx/core/interpreters/inverse_test.py
oryx/core/interpreters/log_prob.py
oryx/core/interpreters/log_prob_test.py
oryx/core/interpreters/propagate.py
oryx/core/interpreters/propagate_test.py
oryx/core/interpreters/unzip.py
oryx/core/interpreters/unzip_test.py
oryx/core/ppl/__init__.py
oryx/core/ppl/transformations.py
oryx/core/ppl/transformations_test.py
oryx/core/state/__init__.py
oryx/core/state/api.py
oryx/core/state/function.py
oryx/core/state/function_test.py
oryx/core/state/module.py
oryx/core/state/registrations.py
oryx/core/state/registrations_test.py
oryx/distributions/__init__.py
oryx/distributions/distribution_extensions.py
oryx/distributions/distribution_extensions_test.py
oryx/experimental/__init__.py
oryx/experimental/mcmc/__init__.py
oryx/experimental/mcmc/kernels.py
oryx/experimental/mcmc/kernels_test.py
oryx/experimental/mcmc/utils.py
oryx/experimental/nn/__init__.py
oryx/experimental/nn/base.py
oryx/experimental/nn/base_test.py
oryx/experimental/nn/combinator.py
oryx/experimental/nn/combinator_test.py
oryx/experimental/nn/convolution.py
oryx/experimental/nn/convolution_test.py
oryx/experimental/nn/core.py
oryx/experimental/nn/core_test.py
oryx/experimental/nn/function.py
oryx/experimental/nn/function_test.py
oryx/experimental/nn/normalization.py
oryx/experimental/nn/normalization_test.py
oryx/experimental/nn/pooling.py
oryx/experimental/nn/pooling_test.py
oryx/experimental/nn/reshape.py
oryx/experimental/nn/reshape_test.py
oryx/experimental/optimizers/__init__.py
oryx/experimental/optimizers/optix.py
oryx/experimental/optimizers/optix_test.py