jax

  • 5th August 2025

saex: training SAEs in Jax

interpretability jax sae

A library for training Sparse Autoencoders in Jax

Read more →
  • 3rd August 2025

jif: Jax text diffusion

jax diffusion

Toy implementation of masked diffusion language models

Read more →