Optax

1 long-form post on Optax: machine-learning research by Taha Bouhsine, each built around live, in-browser interactive visualizations.

1 post tagged #optax.

May 26, 2026

Organizing Randomness: Contrastive Learning in JAX

A block-by-block JAX + Optax implementation of six contrastive losses, each watched as a real animated GIF turning random 2D points into organized embeddings. The runnable companion to "Untangling the Moons."