jax

00380b7e - Merge pull request #37759 from samanklesaria:beta_arg_check