├── .gitignore ├── .pylintrc ├── .style.yapf ├── .vscode └── settings.json ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── oryx ├── __init__.py ├── bijectors │ └── __init__.py ├── core │ ├── __init__.py │ ├── interpreters │ │ ├── __init__.py │ │ ├── harvest.py │ │ ├── harvest_test.py │ │ ├── inverse │ │ │ ├── __init__.py │ │ │ ├── bijector_extensions.py │ │ │ ├── bijector_extensions_test.py │ │ │ ├── core.py │ │ │ ├── custom_inverse.py │ │ │ ├── custom_inverse_test.py │ │ │ ├── inverse_test.py │ │ │ ├── rules.py │ │ │ ├── slice.py │ │ │ └── slice_test.py │ │ ├── log_prob.py │ │ ├── log_prob_test.py │ │ ├── propagate.py │ │ └── propagate_test.py │ ├── kwargs_util.py │ ├── kwargs_util_test.py │ ├── ppl │ │ ├── __init__.py │ │ ├── effect_handler.py │ │ ├── effect_handler_test.py │ │ ├── plate_util.py │ │ ├── transformations.py │ │ └── transformations_test.py │ ├── primitive.py │ ├── primitive_test.py │ ├── pytree.py │ ├── serialize.py │ ├── serialize_test.py │ ├── state │ │ ├── __init__.py │ │ ├── api.py │ │ ├── function.py │ │ ├── function_test.py │ │ ├── module.py │ │ ├── registrations.py │ │ └── registrations_test.py │ └── trace_util.py ├── distributions │ ├── __init__.py │ ├── distribution_extensions.py │ └── distribution_extensions_test.py ├── examples │ └── notebooks │ │ ├── a_tour_of_oryx.ipynb │ │ └── probabilistic_programming.ipynb ├── experimental │ ├── __init__.py │ ├── autoconj │ │ ├── addn.py │ │ ├── addn_test.py │ │ ├── canonicalize.py │ │ ├── canonicalize_test.py │ │ ├── einsum.py │ │ └── einsum_test.py │ ├── matching │ │ ├── __init__.py │ │ ├── jax_rewrite.py │ │ ├── jax_rewrite_test.py │ │ ├── matcher.py │ │ ├── matcher_test.py │ │ ├── rules.py │ │ └── rules_test.py │ ├── mcmc │ │ ├── __init__.py │ │ ├── kernels.py │ │ ├── kernels_test.py │ │ └── utils.py │ ├── nn │ │ ├── __init__.py │ │ ├── base.py │ │ ├── base_test.py │ │ ├── combinator.py │ │ ├── combinator_test.py │ │ ├── convolution.py │ │ ├── convolution_test.py │ │ ├── core.py │ │ ├── core_test.py │ │ ├── function.py │ │ ├── function_test.py │ │ ├── normalization.py │ │ ├── normalization_test.py │ │ ├── pooling.py │ │ ├── pooling_test.py │ │ ├── reshape.py │ │ └── reshape_test.py │ └── optimizers │ │ ├── __init__.py │ │ ├── optix.py │ │ └── optix_test.py ├── internal │ ├── __init__.py │ └── test_util.py ├── tools │ └── build_oryx_docs.py ├── util │ ├── __init__.py │ ├── summary.py │ └── summary_test.py └── version.py └── pyproject.toml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/.pylintrc -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/.style.yapf -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/README.md -------------------------------------------------------------------------------- /oryx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/__init__.py -------------------------------------------------------------------------------- /oryx/bijectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/bijectors/__init__.py -------------------------------------------------------------------------------- /oryx/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/__init__.py -------------------------------------------------------------------------------- /oryx/core/interpreters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/__init__.py -------------------------------------------------------------------------------- /oryx/core/interpreters/harvest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/harvest.py -------------------------------------------------------------------------------- /oryx/core/interpreters/harvest_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/harvest_test.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/__init__.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/bijector_extensions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/bijector_extensions.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/bijector_extensions_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/bijector_extensions_test.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/core.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/custom_inverse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/custom_inverse.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/custom_inverse_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/custom_inverse_test.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/inverse_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/inverse_test.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/rules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/rules.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/slice.py -------------------------------------------------------------------------------- /oryx/core/interpreters/inverse/slice_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/inverse/slice_test.py -------------------------------------------------------------------------------- /oryx/core/interpreters/log_prob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/log_prob.py -------------------------------------------------------------------------------- /oryx/core/interpreters/log_prob_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/log_prob_test.py -------------------------------------------------------------------------------- /oryx/core/interpreters/propagate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/propagate.py -------------------------------------------------------------------------------- /oryx/core/interpreters/propagate_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/interpreters/propagate_test.py -------------------------------------------------------------------------------- /oryx/core/kwargs_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/kwargs_util.py -------------------------------------------------------------------------------- /oryx/core/kwargs_util_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/kwargs_util_test.py -------------------------------------------------------------------------------- /oryx/core/ppl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/ppl/__init__.py -------------------------------------------------------------------------------- /oryx/core/ppl/effect_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/ppl/effect_handler.py -------------------------------------------------------------------------------- /oryx/core/ppl/effect_handler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/ppl/effect_handler_test.py -------------------------------------------------------------------------------- /oryx/core/ppl/plate_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/ppl/plate_util.py -------------------------------------------------------------------------------- /oryx/core/ppl/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/ppl/transformations.py -------------------------------------------------------------------------------- /oryx/core/ppl/transformations_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/ppl/transformations_test.py -------------------------------------------------------------------------------- /oryx/core/primitive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/primitive.py -------------------------------------------------------------------------------- /oryx/core/primitive_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/primitive_test.py -------------------------------------------------------------------------------- /oryx/core/pytree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/pytree.py -------------------------------------------------------------------------------- /oryx/core/serialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/serialize.py -------------------------------------------------------------------------------- /oryx/core/serialize_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/serialize_test.py -------------------------------------------------------------------------------- /oryx/core/state/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/state/__init__.py -------------------------------------------------------------------------------- /oryx/core/state/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/state/api.py -------------------------------------------------------------------------------- /oryx/core/state/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/state/function.py -------------------------------------------------------------------------------- /oryx/core/state/function_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/state/function_test.py -------------------------------------------------------------------------------- /oryx/core/state/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/state/module.py -------------------------------------------------------------------------------- /oryx/core/state/registrations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/state/registrations.py -------------------------------------------------------------------------------- /oryx/core/state/registrations_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/state/registrations_test.py -------------------------------------------------------------------------------- /oryx/core/trace_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/core/trace_util.py -------------------------------------------------------------------------------- /oryx/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/distributions/__init__.py -------------------------------------------------------------------------------- /oryx/distributions/distribution_extensions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/distributions/distribution_extensions.py -------------------------------------------------------------------------------- /oryx/distributions/distribution_extensions_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/distributions/distribution_extensions_test.py -------------------------------------------------------------------------------- /oryx/examples/notebooks/a_tour_of_oryx.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/examples/notebooks/a_tour_of_oryx.ipynb -------------------------------------------------------------------------------- /oryx/examples/notebooks/probabilistic_programming.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/examples/notebooks/probabilistic_programming.ipynb -------------------------------------------------------------------------------- /oryx/experimental/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/__init__.py -------------------------------------------------------------------------------- /oryx/experimental/autoconj/addn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/autoconj/addn.py -------------------------------------------------------------------------------- /oryx/experimental/autoconj/addn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/autoconj/addn_test.py -------------------------------------------------------------------------------- /oryx/experimental/autoconj/canonicalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/autoconj/canonicalize.py -------------------------------------------------------------------------------- /oryx/experimental/autoconj/canonicalize_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/autoconj/canonicalize_test.py -------------------------------------------------------------------------------- /oryx/experimental/autoconj/einsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/autoconj/einsum.py -------------------------------------------------------------------------------- /oryx/experimental/autoconj/einsum_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/autoconj/einsum_test.py -------------------------------------------------------------------------------- /oryx/experimental/matching/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/matching/__init__.py -------------------------------------------------------------------------------- /oryx/experimental/matching/jax_rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/matching/jax_rewrite.py -------------------------------------------------------------------------------- /oryx/experimental/matching/jax_rewrite_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/matching/jax_rewrite_test.py -------------------------------------------------------------------------------- /oryx/experimental/matching/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/matching/matcher.py -------------------------------------------------------------------------------- /oryx/experimental/matching/matcher_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/matching/matcher_test.py -------------------------------------------------------------------------------- /oryx/experimental/matching/rules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/matching/rules.py -------------------------------------------------------------------------------- /oryx/experimental/matching/rules_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/matching/rules_test.py -------------------------------------------------------------------------------- /oryx/experimental/mcmc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/mcmc/__init__.py -------------------------------------------------------------------------------- /oryx/experimental/mcmc/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/mcmc/kernels.py -------------------------------------------------------------------------------- /oryx/experimental/mcmc/kernels_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/mcmc/kernels_test.py -------------------------------------------------------------------------------- /oryx/experimental/mcmc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/mcmc/utils.py -------------------------------------------------------------------------------- /oryx/experimental/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/__init__.py -------------------------------------------------------------------------------- /oryx/experimental/nn/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/base.py -------------------------------------------------------------------------------- /oryx/experimental/nn/base_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/base_test.py -------------------------------------------------------------------------------- /oryx/experimental/nn/combinator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/combinator.py -------------------------------------------------------------------------------- /oryx/experimental/nn/combinator_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/combinator_test.py -------------------------------------------------------------------------------- /oryx/experimental/nn/convolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/convolution.py -------------------------------------------------------------------------------- /oryx/experimental/nn/convolution_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/convolution_test.py -------------------------------------------------------------------------------- /oryx/experimental/nn/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/core.py -------------------------------------------------------------------------------- /oryx/experimental/nn/core_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/core_test.py -------------------------------------------------------------------------------- /oryx/experimental/nn/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/function.py -------------------------------------------------------------------------------- /oryx/experimental/nn/function_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/function_test.py -------------------------------------------------------------------------------- /oryx/experimental/nn/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/normalization.py -------------------------------------------------------------------------------- /oryx/experimental/nn/normalization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/normalization_test.py -------------------------------------------------------------------------------- /oryx/experimental/nn/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/pooling.py -------------------------------------------------------------------------------- /oryx/experimental/nn/pooling_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/pooling_test.py -------------------------------------------------------------------------------- /oryx/experimental/nn/reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/reshape.py -------------------------------------------------------------------------------- /oryx/experimental/nn/reshape_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/nn/reshape_test.py -------------------------------------------------------------------------------- /oryx/experimental/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/optimizers/__init__.py -------------------------------------------------------------------------------- /oryx/experimental/optimizers/optix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/optimizers/optix.py -------------------------------------------------------------------------------- /oryx/experimental/optimizers/optix_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/experimental/optimizers/optix_test.py -------------------------------------------------------------------------------- /oryx/internal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/internal/__init__.py -------------------------------------------------------------------------------- /oryx/internal/test_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/internal/test_util.py -------------------------------------------------------------------------------- /oryx/tools/build_oryx_docs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/tools/build_oryx_docs.py -------------------------------------------------------------------------------- /oryx/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/util/__init__.py -------------------------------------------------------------------------------- /oryx/util/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/util/summary.py -------------------------------------------------------------------------------- /oryx/util/summary_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/util/summary_test.py -------------------------------------------------------------------------------- /oryx/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/oryx/version.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/oryx/HEAD/pyproject.toml --------------------------------------------------------------------------------