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(jax/array-api): dpa1 #4160

Open
wants to merge 14 commits into
base: devel
Choose a base branch
from
Open

feat(jax/array-api): dpa1 #4160

wants to merge 14 commits into from
+1,024 −173

Commits on Sep 24, 2024

  1. feat(jax/array-api): dpa1 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    cad9034 View commit details
    Browse the repository at this point in the history

  2. Apply suggestions from code review 

    Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz @coderabbitai
    njzjz and coderabbitai[bot] authored Sep 24, 2024
    Configuration menu
    Copy the full SHA
    801f36e View commit details
    Browse the repository at this point in the history

  3. Fix test_type_embedding.py 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz authored Sep 24, 2024
    Configuration menu
    Copy the full SHA
    298ed95 View commit details
    Browse the repository at this point in the history

  4. [pre-commit.ci] auto fixes from pre-commit.com hooks 

    for more information, see https://pre-commit.ci
    @pre-commit-ci
    pre-commit-ci[bot] committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    b3f0dd7 View commit details
    Browse the repository at this point in the history

  5. cell may be none

    @njzjz
    njzjz committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    be6a8dc View commit details
    Browse the repository at this point in the history

  6. fix nlist_masked for multiple frames 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    7de9ee3 View commit details
    Browse the repository at this point in the history

Commits on Sep 25, 2024

  1. use a Python implementation of take_along_axis 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 25, 2024
    Configuration menu
    Copy the full SHA
    d65206f View commit details
    Browse the repository at this point in the history

  2. fix zero shape 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 25, 2024
    Configuration menu
    Copy the full SHA
    aff0b42 View commit details
    Browse the repository at this point in the history

  3. fix reshape 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 25, 2024
    Configuration menu
    Copy the full SHA
    e7aeca0 View commit details
    Browse the repository at this point in the history

Commits on Sep 29, 2024

  1. apply flax.nnx.Module 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 29, 2024
    Configuration menu
    Copy the full SHA
    bac980e View commit details
    Browse the repository at this point in the history

  2. fix metaclass 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 29, 2024
    Configuration menu
    Copy the full SHA
    723d6ed View commit details
    Browse the repository at this point in the history

  3. set w, b, idt to Param 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 29, 2024
    Configuration menu
    Copy the full SHA
    d159ffd View commit details
    Browse the repository at this point in the history

  4. __array__ 

    Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
    @njzjz
    njzjz committed Sep 29, 2024
    Configuration menu
    Copy the full SHA
    13e85f7 View commit details
    Browse the repository at this point in the history

  5. revert

    @njzjz
    njzjz authored Sep 29, 2024
    Configuration menu
    Copy the full SHA
    d007f81 View commit details
    Browse the repository at this point in the history