Code

Added a NTXent loss#897

Code

Merged

copybara-service[bot] merged 25 commits intogoogle-deepmind:maingoogle-deepmind/optax:mainfrom GrantMcConachie:patch-1GrantMcConachie/optax:patch-1Copy head branch name to clipboardApr 15, 2024

+145-1

Commits

Commits on Mar 27, 2024

• Created a NTXent loss

  Show description for e2bd891
  GrantMcConachie

  authored
  e2bd891

  Copy full SHA for e2bd891

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

  GrantMcConachie

  authored
  a8f9816

  Copy full SHA for a8f9816

• Update _classification.py

  Show description for b2179c3
  GrantMcConachie

  authored
  b2179c3

  Copy full SHA for b2179c3

• Update _classification_test.py

  Show description for f6599b8
  GrantMcConachie

  authored
  f6599b8

  Copy full SHA for f6599b8

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

  GrantMcConachie

  authored
  9e1908a

  Copy full SHA for 9e1908a

• Update _classification.py

  Show description for 271ccb0
  GrantMcConachie

  authored
  271ccb0

  Copy full SHA for 271ccb0

• Update _classification_test.py

  Show description for 6ae70bf
  GrantMcConachie

  authored
  6ae70bf

  Copy full SHA for 6ae70bf

Commits on Apr 1, 2024

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

  GrantMcConachie

  authored
  b5dbb93

  Copy full SHA for b5dbb93

• Update _classification.py

  Show description for cc79720
  GrantMcConachie

  authored
  cc79720

  Copy full SHA for cc79720

• Update _classification_test.py

  Show description for 4d4368a
  GrantMcConachie

  authored
  4d4368a

  Copy full SHA for 4d4368a

• Update _classification.py

  Show description for 28f86eb
  GrantMcConachie

  authored
  28f86eb

  Copy full SHA for 28f86eb

Commits on Apr 2, 2024

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

  GrantMcConachie

  authored
  45fbefa

  Copy full SHA for 45fbefa

• added contrastive loss specific scripts

  GrantMcConachie

  committed
  2900b7f

  Copy full SHA for 2900b7f

• moved ntxent loss to contrastive specific script and added to __init__

  GrantMcConachie

  committed
  d0a4ce1

  Copy full SHA for d0a4ce1

• added whitespace

  GrantMcConachie

  committed
  8ee73e7

  Copy full SHA for 8ee73e7

Commits on Apr 3, 2024

• changed the file names and fixed the docstrings for the functions

  GrantMcConachie

  committed
  39928cc

  Copy full SHA for 39928cc

• added ntxent to losses.rst and changed import in __init__

  GrantMcConachie

  committed
  4f05cc2

  Copy full SHA for 4f05cc2

• typo fixed

  GrantMcConachie

  committed
  1d8e47f

  Copy full SHA for 1d8e47f

Commits on Apr 4, 2024

• changed typehints

  GrantMcConachie

  committed
  35f3347

  Copy full SHA for 35f3347

• changed back type hints

  GrantMcConachie

  committed
  284060d

  Copy full SHA for 284060d

Commits on Apr 12, 2024

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

  GrantMcConachie

  authored
  4c2d1b3

  Copy full SHA for 4c2d1b3

• another test case that is important to have

  GrantMcConachie

  committed
  a651bcb

  Copy full SHA for a651bcb

• changed the type hints and added a more robust numerical stability calculation

  GrantMcConachie

  committed
  38afc10

  Copy full SHA for 38afc10

• took out trailing whitespace

  GrantMcConachie

  committed
  f66718c

  Copy full SHA for f66718c

Commits on Apr 13, 2024

• minor changes to comments, docstrings, placements

  GrantMcConachie

  committed
  1cef8b5

  Copy full SHA for 1cef8b5