A key aim of the restrictions is to keep AI from supercharging China's military capabilities