Code

[Triplet Margin Loss] Issue 1118#1120

Code

Merged

copybara-service[bot] merged 61 commits intogoogle-deepmind:maingoogle-deepmind/optax:mainfrom cvnad1:issue-1118cvnad1/optax:issue-1118Copy head branch name to clipboardJan 11, 2025

+130-5

Commits

Commits on Oct 26, 2024

• Triplet Marging Loss added

  cvnad1

  committed
  e9811d0

  Copy full SHA for e9811d0

• Triplet Marging Loss added changes

  cvnad1

  committed
  c19ceb3

  Copy full SHA for c19ceb3

• Added Triplet Margin Loss tests

  cvnad1

  committed
  c8f6937

  Copy full SHA for c8f6937

• added spaces

  cvnad1

  committed
  34a8610

  Copy full SHA for 34a8610

• indentation changes

  cvnad1

  committed
  1778d05

  Copy full SHA for 1778d05

• Reduced Indentation and line lengths

  cvnad1

  committed
  5ab74eb

  Copy full SHA for 5ab74eb

• corrected apostrophe marks

  cvnad1

  committed
  510d1c6

  Copy full SHA for 510d1c6

• corrected test file conventions

  cvnad1

  committed
  f4f93c3

  Copy full SHA for f4f93c3

Commits on Nov 2, 2024

• Triplet loss changes

  cvnad1

  committed
  52498f5

  Copy full SHA for 52498f5

• tests

  cvnad1

  committed
  ad40005

  Copy full SHA for ad40005

• Added jit vmap compatibility

  Saanidhyavats

  committed
  ce16c2a

  Copy full SHA for ce16c2a

• Standardized the code

  Saanidhyavats

  committed
  85efeb8

  Copy full SHA for 85efeb8

• Minor correction

  Saanidhyavats

  committed
  a15a2c7

  Copy full SHA for a15a2c7

• tests

  cvnad1

  committed
  fc3c32a

  Copy full SHA for fc3c32a

• modified tests jit vmap

  cvnad1

  committed
  b9f35a5

  Copy full SHA for b9f35a5

• whitespace cleared

  cvnad1

  committed
  1d18f1c

  Copy full SHA for 1d18f1c

• changed dim to shape

  cvnad1

  committed
  b235710

  Copy full SHA for b235710

• changed dim to shape

  cvnad1

  committed
  f36416e

  Copy full SHA for f36416e

• changed vmap and jit

  cvnad1

  committed
  490d941

  Copy full SHA for 490d941

• removed unused packages

  cvnad1

  committed
  7d1b43b

  Copy full SHA for 7d1b43b

Commits on Nov 3, 2024

• tests modified args

  cvnad1

  committed
  6a3a9a1

  Copy full SHA for 6a3a9a1

• testing loss function

  cvnad1

  committed
  80c95fb

  Copy full SHA for 80c95fb

• testing loss function

  cvnad1

  committed
  7f877ef

  Copy full SHA for 7f877ef

• testing loss function

  cvnad1

  committed
  77f6b33

  Copy full SHA for 77f6b33

• Made changes to pass all the tests

  Saanidhyavats

  committed
  6b92b6c

  Copy full SHA for 6b92b6c

• Minor change

  Saanidhyavats

  committed
  7df60cb

  Copy full SHA for 7df60cb

• trailing space

  Saanidhyavats

  committed
  31a0ece

  Copy full SHA for 31a0ece

Commits on Nov 4, 2024

• refactored the code

  cvnad1

  committed
  894dfa2

  Copy full SHA for 894dfa2

• Added triplet_margin_loss in api docs

  Saanidhyavats

  committed
  6bf8aa5

  Copy full SHA for 6bf8aa5

• refactored the code

  cvnad1

  committed
  63a91f6

  Copy full SHA for 63a91f6

• Merge remote-tracking branch 'origin/issue-1118' into issue-1118

  Saanidhyavats

  committed
  9df7799

  Copy full SHA for 9df7799

• refactored the code

  cvnad1

  committed
  e586719

  Copy full SHA for e586719

• Merge branch 'issue-1118' of https://github.com/cvnad1/optax into issue-1118

  cvnad1

  committed
  a7dd576

  Copy full SHA for a7dd576

• Minor correction

  Saanidhyavats

  committed
  cc8377d

  Copy full SHA for cc8377d

• Minor correction

  Saanidhyavats

  committed
  afec78f

  Copy full SHA for afec78f

• Minor correction

  Saanidhyavats

  committed
  4d8f50f

  Copy full SHA for 4d8f50f

• Trailing Spaces

  Saanidhyavats

  committed
  3125b82

  Copy full SHA for 3125b82

• Trailing Spaces

  Saanidhyavats

  committed
  8706919

  Copy full SHA for 8706919

• Trailing Spaces at line 108

  Saanidhyavats

  committed
  954d27e

  Copy full SHA for 954d27e

• Removed jit

  Saanidhyavats

  committed
  fa6e7a7

  Copy full SHA for fa6e7a7

Commits on Nov 7, 2024

• Documentation

  Saanidhyavats

  committed
  35b3e50

  Copy full SHA for 35b3e50

• Testing

  Saanidhyavats

  committed
  a56c0db

  Copy full SHA for a56c0db

Commits on Nov 10, 2024

• Changed the function code and added test accordingly.

  Saanidhyavats

  committed
  b46f945

  Copy full SHA for b46f945

• Merge branch 'main' into issue-1118

  Saanidhyavats

  authored
  12a1efa

  Copy full SHA for 12a1efa

• pylint correction

  Saanidhyavats

  committed
  d91ecc7

  Copy full SHA for d91ecc7

• Merge remote-tracking branch 'origin/issue-1118' into issue-1118

  Saanidhyavats

  committed
  61fa085

  Copy full SHA for 61fa085

• pylint correction __init__.py

  Saanidhyavats

  committed
  ccd3ce5

  Copy full SHA for ccd3ce5

• Added assertion for float

  Saanidhyavats

  committed
  2f9a138

  Copy full SHA for 2f9a138

• place

  cvnad1

  committed
  b4ed379

  Copy full SHA for b4ed379

• Merge branch 'issue-1118' of https://github.com/cvnad1/optax into issue-1118

  cvnad1

  committed
  75183f1

  Copy full SHA for 75183f1

Commits on Nov 11, 2024

• function arguments

  Saanidhyavats

  committed
  6a5e2f2

  Copy full SHA for 6a5e2f2

• minor correction

  Saanidhyavats

  committed
  8d78cf1

  Copy full SHA for 8d78cf1

Commits on Nov 12, 2024

• Merge branch 'google-deepmind:main' into issue-1118

  Saanidhyavats

  authored
  9a641cc

  Copy full SHA for 9a641cc

Commits on Nov 13, 2024

• Merge branch 'issue-1118' of https://github.com/cvnad1/optax into issue-1118

  cvnad1

  committed
  ec34333

  Copy full SHA for ec34333

• Added Docstring

  cvnad1

  committed
  af1bdb2

  Copy full SHA for af1bdb2

• parameterized test

  Saanidhyavats

  committed
  4b4a36e

  Copy full SHA for 4b4a36e

• Merge remote-tracking branch 'origin/issue-1118' into issue-1118

  Saanidhyavats

  committed
  4ddb7b8

  Copy full SHA for 4ddb7b8

Commits on Dec 13, 2024

• Merge branch 'google-deepmind:main' into issue-1118

  cvnad1

  authored
  48d8e0c

  Copy full SHA for 48d8e0c

• changes in formatting

  cvnad1

  committed
  791a899

  Copy full SHA for 791a899

• Completed Requested Changes

  cvnad1

  committed
  3b9a71a

  Copy full SHA for 3b9a71a

• Pylint corrections

  Saanidhyavats

  committed
  d1fb2c7

  Copy full SHA for d1fb2c7