Correct, the Nature Index is tracking high-quality research outputs.

China is also considerably outpacing US in terms of the number of STEM grauduates:

https://cset.georgetown.edu/article/the-global-distribution-of-stem-graduates-which-countries-lead-the-way/