search

cgarciae / treex

A Pytree Module system for Deep Learning in JAX
https://cgarciae.github.io/treex/
MIT License
215 stars 17 forks source link

Develop cosine similarity metric #36

Closed anvelezec closed 2 years ago

anvelezec commented 2 years ago

Develop cosine similarity metric and provide unit test for the implementation following contributes guide.