Sharpened Cosine Similarity in JAX by Raphael Pisoni

colab.research.google.comsite

A JAX/Flax implementation of the Sharpened Cosine Similarity layer.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲

   BlackJAX github.com

   Library of samplers for JAX.

   tool link ↗
   
2. ▲

   Jax in Action manning.com

   A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.

   site topic ai link ↗
   site
3. ▲

   Plugging Into JAX by Nick Doiron medium.com

   Compares Flax, Haiku, and Objax on the Kaggle flower classification challenge.

   site link ↗
   site
4. ▲

   JAXFit github.com

   Accelerated curve fitting library for nonlinear least-squares problems (see arXiv paper).

   tool link ↗
   
5. ▲

   FLAX github.com

   A neural network library for JAX that is designed for flexibility.

   tool topic ai link ↗
   tool
6. ▲

   efax github.com

   Exponential Families in JAX.

   tool link ↗
   

JAX › Models and Projects > Flax: “A JAX/Flax implementation of the Sharpened Cosine Similarity layer.”