Tag AI for climate modeling