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

Enable sampling in chunks with external jax samplers #7465

Open
wants to merge 19 commits into
base: main
Choose a base branch
from
Open

Enable sampling in chunks with external jax samplers #7465

wants to merge 19 commits into from

Commits on Aug 12, 2024

  1. chunking jax samplers

    @andrewdipper
    andrewdipper committed Aug 12, 2024
    Configuration menu
    Copy the full SHA
    ebe4dee View commit details
    Browse the repository at this point in the history

  2. fix print

    @andrewdipper
    andrewdipper committed Aug 12, 2024
    Configuration menu
    Copy the full SHA
    e364b1c View commit details
    Browse the repository at this point in the history

Commits on Aug 13, 2024

  1. add test

    @andrewdipper
    andrewdipper committed Aug 13, 2024
    Configuration menu
    Copy the full SHA
    c771598 View commit details
    Browse the repository at this point in the history

  2. add test

    @andrewdipper
    andrewdipper committed Aug 13, 2024
    Configuration menu
    Copy the full SHA
    44dfc88 View commit details
    Browse the repository at this point in the history

  3. add numpyro test

    @andrewdipper
    andrewdipper committed Aug 13, 2024
    Configuration menu
    Copy the full SHA
    a1b6a9b View commit details
    Browse the repository at this point in the history

Commits on Aug 14, 2024

  1. fix

    @andrewdipper
    andrewdipper committed Aug 14, 2024
    Configuration menu
    Copy the full SHA
    ff32bac View commit details
    Browse the repository at this point in the history

  2. remove overarching jit

    @andrewdipper
    andrewdipper committed Aug 14, 2024
    Configuration menu
    Copy the full SHA
    190ad0f View commit details
    Browse the repository at this point in the history

Commits on Aug 15, 2024

  1. .

    @andrewdipper
    andrewdipper committed Aug 15, 2024
    Configuration menu
    Copy the full SHA
    0ed0066 View commit details
    Browse the repository at this point in the history

  2. fix jit

    @andrewdipper
    andrewdipper committed Aug 15, 2024
    Configuration menu
    Copy the full SHA
    feb53b7 View commit details
    Browse the repository at this point in the history

  3. fix docstring

    @andrewdipper
    andrewdipper committed Aug 15, 2024
    Configuration menu
    Copy the full SHA
    0afc973 View commit details
    Browse the repository at this point in the history

  4. update docstrings

    @andrewdipper
    andrewdipper committed Aug 15, 2024
    Configuration menu
    Copy the full SHA
    270d0b3 View commit details
    Browse the repository at this point in the history

  5. fix lint

    @andrewdipper
    andrewdipper committed Aug 15, 2024
    Configuration menu
    Copy the full SHA
    9e2362e View commit details
    Browse the repository at this point in the history

Commits on Aug 16, 2024

  1. Merge branch 'pymc-devs:main' into chunking

    @andrewdipper
    andrewdipper authored Aug 16, 2024
    Configuration menu
    Copy the full SHA
    bb40568 View commit details
    Browse the repository at this point in the history

  2. enhance test

    @andrewdipper
    andrewdipper committed Aug 16, 2024
    Configuration menu
    Copy the full SHA
    57e52e3 View commit details
    Browse the repository at this point in the history

Commits on Aug 18, 2024

  1. fix _get_log_likelihood / numpyro rng

    @andrewdipper
    andrewdipper committed Aug 18, 2024
    Configuration menu
    Copy the full SHA
    641e142 View commit details
    Browse the repository at this point in the history

  2. add postprocessing on different device

    @andrewdipper
    andrewdipper committed Aug 18, 2024
    Configuration menu
    Copy the full SHA
    9108c60 View commit details
    Browse the repository at this point in the history

Commits on Oct 7, 2024

  1. Merge branch 'pymc-devs:main' into chunking

    @andrewdipper
    andrewdipper authored Oct 7, 2024
    Configuration menu
    Copy the full SHA
    eeb99a8 View commit details
    Browse the repository at this point in the history

  2. switch to blackjax run_inference_algorithm

    @andrewdipper
    andrewdipper committed Oct 7, 2024
    Configuration menu
    Copy the full SHA
    b34b4cc View commit details
    Browse the repository at this point in the history

  3. fix lint

    @andrewdipper
    andrewdipper committed Oct 7, 2024
    Configuration menu
    Copy the full SHA
    4e6b1fa View commit details
    Browse the repository at this point in the history