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

Fix batch size calculation with vmap and static_argnums #1150

Merged
merged 5 commits into from
Sep 24, 2024
Merged

Fix batch size calculation with vmap and static_argnums #1150

merged 5 commits into from
Sep 24, 2024

Commits on Sep 23, 2024

  1. Do not assume arguments are shaped

    @dime10
    dime10 committed Sep 23, 2024
    Configuration menu
    Copy the full SHA
    9132d59 View commit details
    Browse the repository at this point in the history

  2. Improve error message for in_axes values

    @dime10
    dime10 committed Sep 23, 2024
    Configuration menu
    Copy the full SHA
    5754b87 View commit details
    Browse the repository at this point in the history

  3. linting

    @dime10
    dime10 committed Sep 23, 2024
    Configuration menu
    Copy the full SHA
    a40eb46 View commit details
    Browse the repository at this point in the history

  4. changelog

    @dime10
    dime10 committed Sep 23, 2024
    Configuration menu
    Copy the full SHA
    ea868e9 View commit details
    Browse the repository at this point in the history

Commits on Sep 24, 2024

  1. Apply suggestions from code review 

    Co-authored-by: Joey Carter <joseph.carter@xanadu.ai>
    @dime10 @joeycarter
    dime10 and joeycarter authored Sep 24, 2024
    Configuration menu
    Copy the full SHA
    8b23dda View commit details
    Browse the repository at this point in the history