Blog Archive

• Sep 12, 2025 • jax, transformers

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

• Aug 20, 2025 • jax

  SPMD in JAX #1: Sharding