The company, which helps divide AI tasks among different chips, raised $300 million in its latest round