Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Speedup sample and allow specifying compile_kwargs (several major changes related to step samplers) #7578

Merged
merged 11 commits into from
Nov 29, 2024
Merged

Speedup sample and allow specifying compile_kwargs (several major changes related to step samplers) #7578

merged 11 commits into from
Nov 29, 2024

Commits on Nov 25, 2024

  1. More strict/explicit signature in step samplers

    @ricardoV94
    ricardoV94 committed Nov 25, 2024
    Configuration menu
    Copy the full SHA
    5b4279b View commit details
    Browse the repository at this point in the history

  2. Traces are already closed on finally

    @ricardoV94
    ricardoV94 committed Nov 25, 2024
    Configuration menu
    Copy the full SHA
    24fbbe4 View commit details
    Browse the repository at this point in the history

Commits on Nov 27, 2024

  1. Don't recompile Ndarray function on trace slicing

    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    e5eacb8 View commit details
    Browse the repository at this point in the history

  2. Avoid input copy in Ndarray fn 

    Also initialize empty trace and set `trust_input=True`
    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    b589ce8 View commit details
    Browse the repository at this point in the history

  3. Reduce attribute accesses on point record

    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    f979bd6 View commit details
    Browse the repository at this point in the history

  4. ValueGradFunction inner function now accepts a raveled input

    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    acf5175 View commit details
    Browse the repository at this point in the history

  5. Cache size in DictToArrayBijection

    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    812d985 View commit details
    Browse the repository at this point in the history

  6. Optimize ArrayStepShared.step

    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    8708f21 View commit details
    Browse the repository at this point in the history

  7. Optimize NUTS

    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    5efe09e View commit details
    Browse the repository at this point in the history

  8. Avoid recompiling initial_point and logp functions in sample 

    Also removes default `model.check_start_vals()`
    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    8ac7108 View commit details
    Browse the repository at this point in the history

  9. Allow passing compile_kwargs to step inner functions

    @ricardoV94
    ricardoV94 committed Nov 27, 2024
    Configuration menu
    Copy the full SHA
    bd232d2 View commit details
    Browse the repository at this point in the history