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

Feat/jaxtyping #746

Merged
merged 26 commits into from
Sep 4, 2024
Merged

Feat/jaxtyping #746

merged 26 commits into from
Sep 4, 2024

Commits on Aug 22, 2024

  1. feat: continue adding jaxtyping

    @db091756
    db091756 committed Aug 22, 2024
    Configuration menu
    Copy the full SHA
    e23b174 View commit details
    Browse the repository at this point in the history

  2. remove files pulled from different PR

    @db091756
    db091756 committed Aug 22, 2024
    Configuration menu
    Copy the full SHA
    a4c4b99 View commit details
    Browse the repository at this point in the history

Commits on Aug 23, 2024

  1. feat: finished typing pass

    @db091756
    db091756 committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    f9871da View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into feat/jaxtyping

    @db091756
    db091756 authored Aug 23, 2024
    Configuration menu
    Copy the full SHA
    3dc5d0b View commit details
    Browse the repository at this point in the history

  3. fix: add back lost ruff disable

    @db091756
    db091756 committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    a07db43 View commit details
    Browse the repository at this point in the history

  4. feat: another jaxtyping pass

    @db091756
    db091756 committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    2c5eb04 View commit details
    Browse the repository at this point in the history

  5. fix: doc build errors

    @db091756
    db091756 committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    8b91eb9 View commit details
    Browse the repository at this point in the history

  6. feat: added back jit

    @db091756
    db091756 committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    aa188a6 View commit details
    Browse the repository at this point in the history

Commits on Aug 29, 2024

  1. responding to reviewer comments

    @db091756
    db091756 committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    abbd5a2 View commit details
    Browse the repository at this point in the history

  2. feat: added further jaxtyping

    @db091756
    db091756 committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    d574777 View commit details
    Browse the repository at this point in the history

  3. fix: replaced n with m in pairwise

    @db091756
    db091756 committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    16a4bc4 View commit details
    Browse the repository at this point in the history

  4. responding to reviewer comments

    @db091756
    db091756 committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    e2a1cf1 View commit details
    Browse the repository at this point in the history

  5. responding to reviewer comments

    @db091756
    db091756 committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    751406e View commit details
    Browse the repository at this point in the history

  6. feat: added overloads the compute mean method

    @db091756
    db091756 committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    82271f9 View commit details
    Browse the repository at this point in the history

  7. Merge branch 'main' into feat/jaxtyping

    @db091756
    db091756 committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    b897254 View commit details
    Browse the repository at this point in the history

Commits on Aug 30, 2024

  1. feat: added ruff rules

    @db091756
    db091756 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    b94f0c4 View commit details
    Browse the repository at this point in the history

  2. feat: added ... to the exclude_also

    @db091756
    db091756 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    419a639 View commit details
    Browse the repository at this point in the history

  3. responding to reviewer comments

    @db091756
    db091756 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    46bb0ea View commit details
    Browse the repository at this point in the history

  4. fix: fix doc build error

    @db091756
    db091756 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    64e7215 View commit details
    Browse the repository at this point in the history

  5. fix: we do not need to do the as_array conversion when the function i… 

    …s JIT wrapped, could be slowing things down
    @db091756
    db091756 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    4041e0f View commit details
    Browse the repository at this point in the history

  6. fix: fixed missed typing

    @db091756
    db091756 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    713ac4c View commit details
    Browse the repository at this point in the history

Commits on Sep 3, 2024

  1. fix: reverted dimensional control in score_matching.py, improved over… 

    …loads for dimension handler
    @db091756
    db091756 committed Sep 3, 2024
    Configuration menu
    Copy the full SHA
    dce12f7 View commit details
    Browse the repository at this point in the history

  2. docs: changed 'd' to 'n' in overloads

    @db091756
    db091756 committed Sep 3, 2024
    Configuration menu
    Copy the full SHA
    46f2c24 View commit details
    Browse the repository at this point in the history

Commits on Sep 4, 2024

  1. Merge branch 'main' into feat/jaxtyping

    @db091756
    db091756 committed Sep 4, 2024
    Configuration menu
    Copy the full SHA
    860ad4b View commit details
    Browse the repository at this point in the history

  2. docs: improved overloads

    @db091756
    db091756 committed Sep 4, 2024
    Configuration menu
    Copy the full SHA
    bf51a87 View commit details
    Browse the repository at this point in the history

  3. tests: add back old tests

    @db091756
    db091756 committed Sep 4, 2024
    Configuration menu
    Copy the full SHA
    c302936 View commit details
    Browse the repository at this point in the history