Archive of category 'jax'

• Aug 31, 2025 • jax, MLsys

  What happens when you donate an argument in jax.jit?

• Mar 31, 2025 • jax, MLsys

  jax.distributed and XLA Coordination Walkthrough