Law firms vie for tech talent in AI race