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

consider value matrix shapes for Jax conversion #10

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

consider value matrix shapes for Jax conversion #10

wants to merge 2 commits into from

Commits on Oct 18, 2023

  1. consider value matrix shapes for Jax conversion 

    For some programs, the value matrix could have a dimension larger than the largest key-query matrix resulting in a compliation error. By considering both ov and qk matrices when padding the Jax model we can resolve this
    @William-Baker
    William-Baker authored Oct 18, 2023
    Configuration menu
    Copy the full SHA
    7d5f893 View commit details
    Browse the repository at this point in the history

Commits on Jan 22, 2024

  1. added ov test cases

    @William-Baker
    William-Baker committed Jan 22, 2024
    Configuration menu
    Copy the full SHA
    98beaa8 View commit details
    Browse the repository at this point in the history