MuJoCo XLA (MJX) is a specialized extension of the MuJoCo physics engine, designed to run simulations on hardware supported by the XLA (Accelerated Linear Algebra) compiler via the JAX framework.