poets-ai / elegy

A High Level API for Deep Learning in JAX
https://poets-ai.github.io/elegy/
MIT License
469 stars 32 forks source link

Develop cosine similarity metric #206

Closed anvelezec closed 2 years ago

anvelezec commented 2 years ago

Develop cosine similarity metric in treex: https://github.com/cgarciae/treex/issues/36