Nvidia is working on a new lineup of artificial-intelligence chips customized for China as the semiconductor giant tries to maintain access to a huge market while adjusting to shifting U.S. regulations.

After U.S. restrictions cut off chips that Nvidia had previously developed for the market, the company told distributors in China early this week that three new chips could become available as soon as the end of this year, according to people briefed on the matter.