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 #7578

Draft
wants to merge 11 commits into
base: main
Choose a base branch
from
Draft

Speedup sample #7578

wants to merge 11 commits into from

Commits on Nov 23, 2024

  1. Don't recompile Ndarray function on trace slicing

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

  2. Traces are already closed on finally

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

Commits on Nov 24, 2024

  1. Avoid input copy in Ndarray fn 

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

  2. Reduce attribute accesses on point record

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

  3. ValueGradFunction inner function now accepts a raveled input

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

  4. Cache size in DictToArrayBijection

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

  5. Optimize ArrayStepShared.step

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

  6. Optimize NUTS

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

  7. Avoid recompiling initial_point and logp functions in sample 

    Also removes costly `model.check_start_vals()`
Also makes all Step arguments expect vars keyword-only
Also set `trust_input=True` for automatically generated logp_dlogp_function
    @ricardoV94
    ricardoV94 committed Nov 24, 2024
    Configuration menu
    Copy the full SHA
    61627c7 View commit details
    Browse the repository at this point in the history

  8. Allow passing compile_kwargs to step inner functions

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

  9. Disable GC on logp_dlogp function 

    Related to pymc-devs#7539
    @ricardoV94
    ricardoV94 committed Nov 24, 2024
    Configuration menu
    Copy the full SHA
    cb8d51e View commit details
    Browse the repository at this point in the history