Uh they'll just claim the AI is being used in another country then?

There's really no way to definitively decide the tax jurisdiction of an AI agent.

Good luck figuring this one out without a tax on unrealized gains in companies and good luck figuring that one out for companies if they can just scheme around it by reinvesting all their profit indefinitely.

A complete ban on AI might be in order.