Archive of category 'transformers'

• Sep 12, 2025 • jax, transformers

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