Diversity Equity and Inclusion at JAX