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

ntxent fix #946

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

ntxent fix #946

wants to merge 17 commits into from

Commits on Apr 23, 2024

  1. Added a way to subvert the problem with NaNs in the loss. This also a… 

    …llows for 0 vector emeddings.
    @GrantMcConachie
    GrantMcConachie committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    b5d77e5 View commit details
    Browse the repository at this point in the history

  2. Added a 0 vector embedding test to make sure this functionality works

    @GrantMcConachie
    GrantMcConachie committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    030ff4e View commit details
    Browse the repository at this point in the history

  3. Merge branch 'google-deepmind:main' into patch-1

    @GrantMcConachie
    GrantMcConachie authored Apr 23, 2024
    Configuration menu
    Copy the full SHA
    6c5a1f5 View commit details
    Browse the repository at this point in the history

  4. took away the replacement of nans and aded a epsilon value to the cos… 

    …ine similarity.
    @GrantMcConachie
    GrantMcConachie committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    75fb5db View commit details
    Browse the repository at this point in the history

  5. Merge branch 'patch-1' of https://github.com/GrantMcConachie/optax in… 

    …to patch-1
    @GrantMcConachie
    GrantMcConachie committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    20f5f38 View commit details
    Browse the repository at this point in the history

  6. spelling fix

    @GrantMcConachie
    GrantMcConachie committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    1358d70 View commit details
    Browse the repository at this point in the history

Commits on Apr 29, 2024

  1. Merge branch 'google-deepmind:main' into patch-1

    @GrantMcConachie
    GrantMcConachie authored Apr 29, 2024
    Configuration menu
    Copy the full SHA
    bb8c208 View commit details
    Browse the repository at this point in the history

Commits on May 6, 2024

  1. Merge branch 'google-deepmind:main' into patch-1

    @GrantMcConachie
    GrantMcConachie authored May 6, 2024
    Configuration menu
    Copy the full SHA
    b3a17bc View commit details
    Browse the repository at this point in the history

  2. Added a doctest

    @GrantMcConachie
    GrantMcConachie committed May 6, 2024
    Configuration menu
    Copy the full SHA
    11ee46b View commit details
    Browse the repository at this point in the history

Commits on May 7, 2024

  1. rewrite docstring to pass syntax checks

    @GrantMcConachie
    GrantMcConachie committed May 7, 2024
    Configuration menu
    Copy the full SHA
    037f492 View commit details
    Browse the repository at this point in the history

  2. truncated decimal points in doctest

    @GrantMcConachie
    GrantMcConachie committed May 7, 2024
    Configuration menu
    Copy the full SHA
    b1d8eb3 View commit details
    Browse the repository at this point in the history

  3. added a space in the print statements for the doctest

    @GrantMcConachie
    GrantMcConachie committed May 7, 2024
    Configuration menu
    Copy the full SHA
    24edd0b View commit details
    Browse the repository at this point in the history

  4. minor fixes to doctest

    @GrantMcConachie
    GrantMcConachie committed May 7, 2024
    Configuration menu
    Copy the full SHA
    bb325d7 View commit details
    Browse the repository at this point in the history

  5. changed rng key

    @GrantMcConachie
    GrantMcConachie committed May 7, 2024
    Configuration menu
    Copy the full SHA
    a98cec8 View commit details
    Browse the repository at this point in the history

Commits on May 31, 2024

  1. changed eposilon to match dtype

    @GrantMcConachie
    GrantMcConachie committed May 31, 2024
    Configuration menu
    Copy the full SHA
    d05337b View commit details
    Browse the repository at this point in the history

Commits on Jun 10, 2024

  1. Merge branch 'google-deepmind:main' into patch-1

    @GrantMcConachie
    GrantMcConachie authored Jun 10, 2024
    Configuration menu
    Copy the full SHA
    11ed8b7 View commit details
    Browse the repository at this point in the history

  2. changed np to jnp to get rid of numpy import

    @GrantMcConachie
    GrantMcConachie committed Jun 10, 2024
    Configuration menu
    Copy the full SHA
    1bf4b2e View commit details
    Browse the repository at this point in the history