Blog Archive

• Oct 20, 2025 • optimization

  A Faster Manifold Muon with ADMM

• Sep 12, 2025 • jax, transformers

  SPMD in JAX #2: Transformers in Bare-Metal JAX

• Aug 20, 2025 • jax

  SPMD in JAX #1: Sharding